/jax_modules

Collection of useful jax snippets and modules to build more complex differentiable code

MIT LicenseMIT

No issues in this repository yet.