pytorch

An open source deep learning platform that provides a seamless path from research prototyping to production deployment.