PyTorch es una biblioteca de aprendizaje profundo de código abierto, desarrollada originalmente por Meta Platforms y actualmente desarrollada con el apoyo de la Linux Foundation. Sucesora de Torch, PyTorch proporciona una API de alto nivel que se basa en implementaciones optimizadas de bajo nivel de algoritmos y arquitecturas de aprendizaje profundo, como el Transformer o SGD. En particular, esta API simplifica el entrenamiento y la inferencia de modelos a unas pocas líneas de código. PyTorch permite la paralelización automática del entrenamiento y, internamente, implementa enlaces CUDA que aceleran aún más el entrenamiento al aprovechar los recursos de GPU.
- Website
- http://pytorch.org, https://pytorch.org