On-the-fly conversions between Jax and NumPy tensors - View it on GitHub