Simple tools to mix and match PyTorch and Jax - Get the best of both worlds! - View it on GitHub