Proof-of-concept of global switching between numpy/jax/pytorch in a library. - View it on GitHub