JAX-based neural network library
Why do you think that https://github.com/patrick-kidger/diffrax is a good alternative to dm-haiku
JAX-based neural network library
Why do you think that https://github.com/patrick-kidger/diffrax is a good alternative to dm-haiku