https://jackd.github.io/posts/deq/
Deep Equilibrium Models in Jax - jackd