OpenAI standardizes on PyTorch

Listen to article

We are standardizing OpenAI’s deep learning framework onPyTorch⁠(opens in a new window). In the past, we implemented projects in many frameworks depending on their relative strengths. We’ve now chosen to standardize to make it easier for our team to create and share optimized implementations of our models.

As part of this move, we’ve just released aPyTorch-enabled version⁠(opens in a new window)ofSpinning Up in Deep RL⁠, an open-source educational resource produced by OpenAI that makes it easier to learn about deep reinforcement learning. We are also in the process of writing PyTorch bindings for our highly-optimizedblocksparse kernels⁠, and will open-source those bindings in upcoming months.

The main reason we’ve chosen PyTorch is to increase our research productivity at scale on GPUs. It is very easy to try and execute new research ideas in PyTorch; for example, switching to PyTorch decreased our iteration time on research ideas in generative modeling from weeks to days. We’re also excited to be joining a rapidly-growing developer community, including organizations like Facebook and Microsoft, in pushing scale and performance on GPUs.

Going forward we’ll primarily use PyTorch as our deep learning framework but sometimes use other ones when there’s a specific technical reason to do so. Many of our teams have already made the switch, and we look forward to contributing to the PyTorch community in upcoming months.

* Framework

* 2020

Author

OpenAI

Related articles

View all

Global news partnerships: Le Monde and Prisa Media Company Mar 13, 2024

OpenAI announces new members to board of directors Company Mar 8, 2024

Review completed & Altman, Brockman to continue to lead OpenAI Company Mar 8, 2024