/apple_m1_pro_python

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

Primary LanguageJupyter Notebook

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.