An open-source machine learning framework used for computer vision and NLP.
Module name
pytorch-gpu
PyTorch is an open-source machine learning framework widely used for deep learning research and production. It provides flexibility and speed when building complex neural networks, especially in research areas such as computer vision and natural language processing. SeaWulf provides multiple PyTorch versions supporting both CPU and GPU configurations.
Applies to
Machine learning & AI