Skip to content

Add JaxModule: enables using jax functions in PyTorch autograd graphs
Compare
Choose a tag to compare
@lebrice lebrice released this 06 Jun 20:55
· 34 commits to master since this release
v0.0.4
754445a
This commit was signed with the committer’s verified signature.
lebrice Fabrice Normandin
GPG key ID: 502F6DFDEC13F08A
Verified
Learn about vigilant mode.

Example: https://github.com/lebrice/torch_jax_interop/blob/754445a1ce9a473b51d126b7919512cda22a0fa6/torch_jax_interop/to_torch_test.py#L279

Full Changelog: v0.0.3...v0.0.4

Assets 2
Loading