apple_m1_pro_python icon indicating copy to clipboard operation
apple_m1_pro_python copied to clipboard

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:

Tensorflow Resnet50:

tf_resnet_50results.png

PyTorch Resnet50:

  • Difference between CPU and GPU gpu_vs_cpu.png
  • Comparing with Nvidia samples_sec.png

PyTorch Bert

  • Running a Bert from Huggingface pt_bert.png

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.