Tensors and Dynamic neural networks in Python with strong GPU acceleration
PyTorch is an open source machine learning framework that accelerates the path from research prototyping to production deployment.
PyTorch has the following features:
- Production Ready: Transition seamlessly between eager and graph modes with TorchScript, and accelerate the path to production with TorchServe.
- Distributed Training: Scalable distributed training and performance optimization in research and production is enabled by the torch.distributed backend.
- Robust Ecosystem: A rich ecosystem of tools and libraries extends PyTorch and supports development in computer vision, NLP and more.er on the server using Node and power mobile apps using React Native.
- Cloud Support: PyTorch is well supported on major cloud platforms, providing frictionless development and easy scaling.
Detectron2 is a platform for object detection, segmentation and other visual recognition tasks.
A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.