3 ms·Trax is built on top of Jax, which is built on top of Autograd, which is a PyTorch like thingby chewxy 7y agoTrax is built on top of Jax, which is built on top of Autograd, which is a PyTorch like thing