GitHub repository: lebrice/torch_jax_interop
Primary language: Python
Created: 2024-05-30T18:45:35Z
Description: Simple tools to mix and match PyTorch and Jax - Get the best of both worlds!