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