You can perform almost all Numpy Operations with Jax. JAX provides 'jax.numpy' that helps you to perform Almost anything that can be done with 'numpy' Here is how you can use "jax.numpy"

→ View original post on X — @sumanth_077,