Argmax
On-device AI Frameworks Engineer (Staff)
Palo Alto, CA +1 · Staff · Full-time · Posted Oct 29, 2025
- Location
- Palo Alto, CA
Palo Alto, California, United States - Experience
- 3+ years required; 5+ years preferred, with the stated platform alternatives.
About the job
Designs and optimizes developer-facing frameworks for running AI inference natively on Apple and Android devices, turning research prototypes into production features and supporting enterprise customers.
What you'll do
- Productionize Python research prototypes through test-driven implementation, experimentation, and benchmarking.
- Recommend scalable SDK and framework designs as supported models and workloads grow.
- Work with enterprise customer engineering teams on onboarding, questions, and feature customization.
What they ask for
- 3+ years of hands-on SDK or frameworks development for iOS or macOS.
- Fluency in Swift.
- Fluency in profiling and optimizing native applications.
- Familiarity with at least one of Core ML, MLX Swift, LiteRT, ONNX, WebGPU or ExecuTorch.
Nice to have
- 5+ years of hands-on SDK or frameworks development for iOS or Android.
- Significant open-source contributions; fluency in Swift, Kotlin and Python; or direct experience with Core ML, MLX Swift and LiteRT.
Tools
Swift, Python, Kotlin, Core ML, MLX Swift, LiteRT, ONNX, WebGPU
AI focus
Builds on-device AI inference frameworks, productionizes research prototypes, and may contribute experimentation and benchmarking toward AI research conference submissions.
Model-written summary · the original posting is authoritative