Train Tensorflow models using Neural Engine on M2 chip

Hi!

I've just got a MacBook Air with M2 chip. I'm doing some research on ML and I was wondering if it is possible to use Neural Engine as an accelerator for training. If it is possible, where can I find ressources on how to enable it and how to check it?

Thank you!

Up vote post of gn48 Down vote post of gn48
2.4k views
Posted by
gn48
Reply
Add a Comment

Accepted Reply

There is a Metal TensorFlow Plugin available, which accelerates model training using your Mac's GPU.

Posted by
CMDdev
Up vote reply of CMDdev Down vote reply of CMDdev
Post marked as solved
Add a Comment

Replies

There is a Metal TensorFlow Plugin available, which accelerates model training using your Mac's GPU.

Posted by
CMDdev
Up vote reply of CMDdev Down vote reply of CMDdev
Post marked as solved
Add a Comment

I saw that but I was wondering if it only use GPU or if it gives us access to the Neural Engine

Posted by
gn48
Up vote reply of gn48 Down vote reply of gn48
Add a Comment

Hi,

I was wondering the same, did you find an answer/solution to that?

Posted by
NFMi
Up vote reply of NFMi Down vote reply of NFMi
Add a Comment