JAX-based neural network library
Why do you think that https://github.com/google/maxtext is a good alternative to dm-haiku
Here you can share your experience with the project you are suggesting or its comparison with dm-haiku. Optional.
A valid email to send you a verification link when necessary or log in.