← All projects

PyTorch

The deep learning community home for the open source PyTorch framework

AI Toolsdeep-learningmachine-learningopen-sourcepythonneural-networksdistributed-traininggpu-computing
PyTorch screenshot

About

PyTorch is an open-source deep learning framework and ecosystem maintained by the PyTorch Foundation. It provides tools for building and deploying machine learning models, including distributed training, TorchScript for production deployment, and a rich ecosystem of libraries for computer vision, NLP, and more. It is widely used in academic research and industry production environments.

Problem

Building, training, and deploying deep learning models requires a flexible yet production-ready framework with broad hardware and cloud support.

For

machine learning researchers and engineers

How it works

PyTorch provides a Python-based tensor computation library with GPU acceleration, automatic differentiation, and tools like TorchScript and TorchServe to transition models from research to production.

Business model

open-source

Status

launched

Company

PyTorch Foundation

Similar projects