lipengyu / apple_m1_pro_python

A collection of ML scripts to test the M1 Pro MacBook Pro

Apple Silicon DL benchmarks

Currently we have PyTorch and Tensorflow that have Metal backend.

Results

Varied results across frameworks:

• Apple M1Pro Pytorch Training Results
• Apple M1Pro Tensorflow Training Results

Tensorflow Resnet50:

PyTorch Resnet50:

• Difference between CPU and GPU
• Comparing with Nvidia

PyTorch Bert

• Running a Bert from Huggingface

Pytorch

We have official PyTorch support! check pytorch folder to start running your benchmarks

Tensorflow

You can run tensorflow benchmarks by going to the tensorflow folder.