Timeline
2024 - 2025
- General model conversion successful
- Stateful model conversion successful
- On-device porting experiments in progress
Tools & skills
Skills
- Core ML / MLCompute
- Swift
- On-device inference pipeline configuration
Motivation
- I wanted to build ML models optimized for Apple CPU, GPU, and NPU that could be used in mobile keyboards.
- I wanted to implement AI-powered autocomplete inference for keyboards, given their limited memory resources and the fact that CoreML doesn't track memory.
What I learned
- Learned how to convert PyTorch models using coremltools.
- Studied how to convert to Stateful models using the KV-cache introduced in iOS 18.