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