PyTorch

An open source machine learning framework that accelerates the path from research prototyping to production deployment.