Skip to Content

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

macbook air blue

What's Coming in the M5 MacBook Air

Thursday February 26, 2026 3:57 pm PST by
Along with the low-cost MacBook, Apple could introduce a refreshed version of the MacBook Air next week. Most of the focus will be on the new machine, but the MacBook Air is expected to get some useful internal updates. M5 Chip The next-generation MacBook Air will adopt the M5 chip, which Apple already introduced in the iPad Pro and MacBook Pro models that came out last year. Apple's M5...
ipad mini 7 1

Could Apple's OLED iPad Mini Finally Be a Kindle Killer?

Friday February 27, 2026 12:08 pm PST by
With a similar screen size and easy, one-handed grip, the iPad mini has always been the Apple device that overlaps most with dedicated e-readers. Now, amid rumors pointing to an OLED display for the next generation, could the iPad mini finally replace devices such as the Kindle and Kobo? The shift from LCD to OLED could make the iPad mini far more appealing as a reading device. OLED panels...
Apple Announces Special Event in New York Feature 1

Apple Teases 'A Big Week Ahead' With Announcements Starting Monday

Thursday February 26, 2026 6:06 am PST by
Apple CEO Tim Cook today teased "a big week ahead," with announcements starting Monday. His post included an #AppleLaunch hashtag with a colorful Apple logo, along with a short video that ultimately shows an Apple logo on the lid of a Mac. Apple is reportedly planning a three-day stretch of product announcements from Monday, March 2 through Wednesday, March 4, with up to five new products...

Top Rated Comments

innominato5090 Avatar
50 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)
50 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
50 months ago
I hope to be blown away with what wwdc has to announce
Score: 4 Votes (Like | Disagree)
adamw Avatar
50 months ago
Always good to see move native support for Metal in MacOS on M1 family of chips...
Score: 3 Votes (Like | Disagree)
50 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
49 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)