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
License
BSD License
You Might Also Like
Rate This Project
Login To Rate This Project
User Reviews
Be the first to post a review of PyTorch!