GitHub repository: Artur-Galstyan/equinox
Primary language: Python
Description: Elegant easy-to-use neural networks in JAX. https://docs.kidger.site/equinox/