apple/coreai-torch

Bridges PyTorch and Core AI. Convert existing models to Core AI IR, or author new ones from PyTorch via composite ops, custom op lowerings, and inline Metal GPU kernels.

语言：Python · GitHub 星标：177 · Fork：51

在 GitHub 上查看原仓库