Simple tools to mix and match PyTorch and Jax - Get the best of both worlds!
Primary LanguagePythonMIT LicenseMIT