equinox
Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/
2k stars
23 watching
136 forks
Language: Python
last commit: 1 day ago
Linked from 1 awesome list
deep-learningequinoxjaxneural-networks
n2cholas/awesome-jax