Yifan Shen

Model export recipes, Python primitives, and Swift runtime utilities for on-device AI

Mathematical & chemical routines e.g. nonlinear optimization etc. with c++ & python frontend

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.

A library for PyTorch model compression and optimizations for deployment via Core AI on Apple silicon.

End-to-end solution for enabling on-device AI across mobile and edge devices for PyTorch models

Core ML tools contain supporting tools for Core ML model conversion, editing, and validation.