Unofficial JAX implementations of deep learning research papers
Why do you think that https://github.com/patrick-kidger/diffrax 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/diffrax is a good alternative to jax-models