lebrice / torch_jax_interop

Simple tools to mix and match PyTorch and Jax - Get the best of both worlds!
26Updated this week

Alternatives and similar repositories for torch_jax_interop:

Users that are interested in torch_jax_interop are comparing it to the libraries listed below