PyTorch is a Python package that offers Tensor computation (like NumPy) with strong GPU acceleration and deep neural networks built on tape-based autograd system. This project allows for fast, flexible experimentation and efficient production.

PyTorch consists of torch (Tensor library), torch.autograd (tape-based automatic differentiation library), torch.jit (a compilation stack [TorchScript]), torch.nn (neural networks library), torch.multiprocessing (Python multiprocessing), and torch.utils (DataLoader and other utility functions). PyTorch can be used as a replacement for Numpy, or as a deep learning research platform that provides optimum flexibility and speed.

Features

  • TorchScript
  • Distributed training
  • Tools and libraries
  • Cloud partners
  • Native ONNX support
  • C++ front-end
  • machine learning framework

Project Samples

Play Video PyTorch Mobile

Project Activity

See All Activity >

Categories

Machine Learning, Deep Learning Frameworks

License

BSD License

Follow PyTorch

PyTorch Web Site

Other Useful Business Software
Gen AI apps are built with MongoDB Atlas Icon
Gen AI apps are built with MongoDB Atlas

The database for AI-powered applications.

MongoDB Atlas is the developer-friendly database used to build, scale, and run gen AI and LLM-powered apps—without needing a separate vector database. Atlas offers built-in vector search, global availability across 115+ regions, and flexible document modeling. Start building AI apps faster, all in one place.
Start Free
Rate This Project
Login To Rate This Project

User Reviews

Be the first to post a review of PyTorch!

Additional Project Details

Operating Systems

Linux, Mac, Windows

Registered

2020-01-17
Report inappropriate content