/flaxfit

Fitting jax flax models made simple.

Primary LanguagePythonMIT LicenseMIT

Watchers