Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/
Why do you think that https://github.com/patrick-kidger/torchtyping is a good alternative to equinox
Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/
Why do you think that https://github.com/patrick-kidger/torchtyping is a good alternative to equinox