Machine Learning Framework PyTorch Enabling GPU-Accelerated Training on Apple Silicon Macs

In collaboration with the Metal engineering team at Apple, PyTorch today announced that its open source machine learning framework will soon support GPU-accelerated model training on Apple silicon Macs powered by M1, M1 Pro, M1 Max, or M1 Ultra chips.

pytorch
Until now, PyTorch training on the Mac only leveraged the CPU, but an upcoming version will allow developers and researchers to take advantage of the integrated GPU in Apple silicon chips for "significantly faster" model training.

A preview build of PyTorch version 1.12 with GPU-accelerated training is available for Apple silicon Macs running macOS 12.3 or later with a native version of Python. Performance comparisons and additional details are available on PyTorch's website.

Tag: Metal

Popular Stories

apple wallet drivers license feature iPhone 15 pro

iPhone Driver's Licenses: These 17 U.S. States Offer Them or Will Later

Thursday June 19, 2025 11:28 am PDT by
In select U.S. states, residents can add their driver's license or state ID to the Wallet app on the iPhone and Apple Watch, providing a convenient and contactless way to display proof of identity or age at select airports and businesses, and in select apps. Unfortunately, this feature continues to roll out very slowly since it was announced in 2021, with only nine U.S. states and Puerto...
ios 26 call holding

iOS 26 Beta is Hiding a New Ringtone — Here's What It Sounds Like

Thursday June 19, 2025 7:25 pm PDT by
Apple is hiding a new ringtone within iOS 26. The new ringtone is an alternative version of the existing Reflection ringtone, which has been the default ringtone since the iPhone X was released in 2017. It was discovered within the code for the first developer beta of iOS 26, but it remains hidden, so you will not find it in the list of ringtones available in the Settings app for now. It...
iPhone 17 Pro Blue Feature Tighter Crop

iPhone 17 Pro Launching in Three Months With These 12 New Features

Saturday June 14, 2025 5:45 pm PDT by
The iPhone 17 Pro and iPhone 17 Pro Max are three months away, and there are plenty of rumors about the devices. Below, we recap key changes rumored for the iPhone 17 Pro models as of June 2025:Aluminum frame: iPhone 17 Pro models are rumored to have an aluminum frame, whereas the iPhone 15 Pro and iPhone 16 Pro models have a titanium frame, and the iPhone X through iPhone 14 Pro have a...
iOS 18

Apple Releases iOS 18.6 Public Beta

Wednesday June 18, 2025 10:24 am PDT by
Apple today seeded the first betas of upcoming iOS 18.6 and iPadOS 18.6 updates to public beta testers, with the betas coming just a few days after Apple provided the betas to developers. Testers who have signed up for beta updates through Apple's beta site can download iOS 18.6 and iPadOS 18.6 from the Settings app on a compatible device by going to General > Software Update. When the...
apple watch ultra 2 new black

Apple Watch Ultra 3 Finally Coming After Two-Year Hiatus

Monday June 16, 2025 8:45 am PDT by
Apple will finally deliver the Apple Watch Ultra 3 sometime this year, according to analyst Jeff Pu of GF Securities Hong Kong (via @jukanlosreve). The analyst expects both the Apple Watch Series 11 and Apple Watch Ultra 3 to arrive this year (likely alongside the new iPhone 17 lineup, if previous launches are anything to go by), according to his latest product roadmap shared with...
iPhone 16 Battery Life Feature

iOS 26's New Battery Life Mode Available Only on These iPhone Models

Saturday June 21, 2025 9:02 am PDT by
Last week, we reported that iOS 26 introduces an opt-in Adaptive Power Mode on the iPhone, alongside the existing Low Power Mode. Apple says that Adaptive Power Mode can make "small performance adjustments" when necessary to extend an iPhone's battery life, including slightly lowering the display brightness or allowing some activities to "take a little longer." The full description of...
General Spotify Feature

Spotify Preparing to Launch Long-Awaited Lossless Audio Tier on iPhone

Thursday June 19, 2025 1:46 pm PDT by
Spotify appears to be gearing up to launch its long-awaited lossless music tier. Chris Messina (via TechCrunch) and Spicetify (via The Verge) spotted new lossless references within the code for Spotify's desktop app and web player. With assistance from Aaron Perris, MacRumors has confirmed that the latest beta of the Spotify app for the iPhone also contains new lossless-related code....
Craig Federighi No

John Gruber Reacts to Apple Declining His Interview After His Criticism

Wednesday June 18, 2025 8:10 pm PDT by
Every year between 2015 and 2024, at least one Apple executive agreed to be interviewed by Daring Fireball's John Gruber for a special WWDC episode of his podcast, The Talk Show. Last year, for example, Apple's software engineering chief Craig Federighi, marketing chief Greg Joswiak, and top AI researcher John Giannandrea joined Gruber on stage at the California Theatre in San Jose to discuss...
iPhone 17 Base Model Rumored to Come in New Green and Purple Colors Feature

iPhone 17 Base Model Rumored to Come in New Purple and Green Colors

Friday June 20, 2025 7:24 am PDT by
Apple is testing new Purple and Green color options for the iPhone 17 base model, according to new information shared by a leaker known as Majin Bu. In a blog post today, Majin Bu said that only one of the two new colors might make the final cut, with Purple apparently being the more likely choice. The base model iPhone 16 is available in five colors: Ultramarine, Teal, Pink, White, and...

Top Rated Comments

innominato5090 Avatar
40 months ago
So unbelievably excited about this. Being able to test GPU-accelerated code locally (and even train some smaller networks) rather than having to rely on a unix-server will speed up ML development significantly.

Can't wait for it to become mature enough to be merged in a stable branch.
Score: 13 Votes (Like | Disagree)
mukiex Avatar
40 months ago
Personally I'm hoping to see support for the neural engine in the future. When they added NE support to Topaz, my M1 Macbook Air suddenly started performing on par with my desktop GTX 1080.
Score: 5 Votes (Like | Disagree)
mrWONERFUL Avatar
40 months ago
I hope to be blown away with what wwdc has to announce
Score: 4 Votes (Like | Disagree)
adamw Avatar
40 months ago
Always good to see move native support for Metal in MacOS on M1 family of chips...
Score: 3 Votes (Like | Disagree)
seek3r Avatar
40 months ago
Nice step in the direction of finding out how capable these GPUs are for this kind of workload, may also give an idea if they work well enough for GPGPU and ML work to infer if Apple is likely to develop their own GPU for the eventual AS Mac Pro
Score: 3 Votes (Like | Disagree)
Luca1995it Avatar
40 months ago
I started collecting benchmarks of M1 Max on PyTorch here: https://github.com/lucadiliello/pytorch-apple-silicon-benchmarks
Score: 2 Votes (Like | Disagree)