3 ms·Pytorch and Jax have a numpy like API, so you can use PyTorch for other things too.by figomore 3y agoPytorch and Jax have a numpy like API, so you can use PyTorch for other things too.