Core AI PyTorch Extensions
Bring PyTorch models to Core AI for on-device execution.
Core AI PyTorch Extensions is Apple's Python package, coreai-torch, for bridging PyTorch and Core AI by converting PyTorch exported programs to Core AI IR/assets and enabling Core AI model authoring with composite ops, custom lowerings, and inline Metal GPU kernels.
Recent stories
0 linked stories
No linked stories yet.