equinox (pypi): Elegant easy-to-use neural networks in JAX.
Keywords: deep-learning|equinox|jax|neural-networks