Why do you think that https://github.com/google/flax is a good alternative to jaxline
Here you can share your experience with the project you are suggesting or its comparison with jaxline. Optional.
A valid email to send you a verification link when necessary or log in.