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