/jaxfi-JAXFriendlyInterface

Friendly Interface to JAX, `jaxfi` simplifies JAX interface to replicate PyTorch's ease of use.

Primary LanguagePythonMIT LicenseMIT

Watchers