PYTORCH IS DEAD. LONG LIVE JAX.

Summary from the TLDR Web Dev newsletter

The PyTorch framework was initially successful for rapid prototyping, but now has a better successor in JAX, a framework created by Google DeepMind. JAX has a compiler-driven approach, which allows for automatic optimization and parallelization. It also has a functional API, better reproducibility, portability across different hardware, and automatic scaling.