Some recent works such as Deep Equilibrium Models 3 use a black-box iterative algorithm that converges to some equilibrium state, and compute gradients at the equilibrium point, without back-propping through all the (arbitrary number of) iterations. What support is there for this in PyTorch?