How to do backward pass when parts of a model are on separate systems?

4

That was simpler than I expected and exactly what I needed. Tested. Compared resulting gradients against the combined model. Perfect match. Thank you.