Skip to content

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

Notifications You must be signed in to change notification settings

tcapelle/apple_m1_pro_python

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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.