/nnx

Neural Networks for JAX

Primary LanguagePythonMIT LicenseMIT

Watchers