Implementing Keras custom layer in Core ML model with enabling GPU acceleration

Code for GPU computation in Apple Core ML model converted from Keras

Takashi Shirakawa
Mac O’Clock
Published in
4 min readMar 8, 2021

--

Backgrounds

I use a Windows machine with nVidia’s powerful GPU for training my Keras model with a huge dataset. This means that I must convert the trained Keras model into Apple’s Core…

--

--

Takashi Shirakawa
Mac O’Clock

Surgeon + Engineer + Programmer / Love eating, outdoor and travel