PyTorch is an open source machine learning framework that accelerates the path from research prototyping to production deployment. PyTorch enables fast, flexible experimentation and efficient production through a user-friendly front-end, distributed training, and ecosystem of tools and libraries.