This repository is work in progress.
Implementation of different machine learning models using Apple's Metal Performance Shaders Graph API.
Model | Implemented | Note |
---|---|---|
VGG16BN | Yes | BN does not use running mean |
ResNet50 | In progress | |
UNet-2D | Planned | |
UNet-3D | Planned |
TODO.
Max-Heinrich Laves
@MaxLaves
@[email protected]