GitHub repository: packquickly/equinox_models
Primary language: Python
Description: A repo for neural networks written in JAX+Equinox