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
Auth for GenAI | Auth0 Icon
Auth for GenAI | Auth0

Enable AI agents to securely access tools, workflows, and data with fine-grained control and just a few lines of code.

Easily implement secure login experiences for AI Agents - from interactive chatbots to background workers with Auth0. Auth for GenAI is now available in Developer Preview
Try free now
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