/jax-loss

A jax library of common machine learning loss functions

Primary LanguagePythonMIT LicenseMIT

Watchers