PyTorch 2.1 ya está disponible, ofreciendo una mejora de 35 veces en la velocidad de compilación y la capacidad de implementar IA en dispositivos móviles a través de ExecuTorch. Esta nueva versión incluye varias mejoras de rendimiento y soporte para la cuantificación. El código NumPy se puede compilar en C++/CUDA usando torch.compile, lo que resulta en un aumento significativo del rendimiento. A pesar del aumento en el uso de Python, la velocidad de ejecución es varias veces más rápida.