/equinox_models

A repo for neural networks written in JAX+Equinox

Primary LanguagePythonApache License 2.0Apache-2.0

EquinoxMonorepo

A repo for neural networks written in JAX+Equinox