Unofficial JAX implementations of deep learning research papers
Why do you think that https://github.com/patrick-kidger/equinox is a good alternative to jax-models
Unofficial JAX implementations of deep learning research papers
Why do you think that https://github.com/patrick-kidger/equinox is a good alternative to jax-models