philipturner / climate-change-forecasting Goto Github PK

Evidence strongly suggests a PyTorch GPU backend will be released at WWDC 2022. Here is the reasoning:

• Initially, Soumith may have accidentally hinted that they are working with Apple. We had assumed that was a reference to collaborating with AMD for ROCm.
• PyTorch engineers do not have experience with Metal. Neither does TensorFlow. The solution: get Apple's assistance.
• PyTorch is separate enough from Facebook that Apple could make an exception. An ML framework will not mine anyone's private data.
• PyTorch's team consistently evades our questions about the release date. They are very diligent about keeping it confidential.
• Finally, from Apple's job request for CPU acceleration (posted March 4, 2022):

You will also collaborate with the PyTorch community to get this work integrated into PyTorch.