GitHub repository: vishalbelsare/equinox
Primary language: Python
Description: Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/