Argmax
On-device AI Frameworks Engineer (Staff)
Manhattan, NY +1 · Staff · Full-time · Posted Oct 29, 2025
- Location
- Manhattan, NY
Palo Alto, California, United States - Experience
- 3+ years required; preferred qualification states 5+ years of hands-on SDK or Frameworks development for iOS or Android.
About the job
Designs, implements, and optimizes frameworks and developer APIs for running AI inference natively on Apple and Android devices.
What you'll do
- Turn Applied Research Python prototypes into production-ready framework implementations, including testing, experimentation, and benchmarking.
- Recommend scalable SDK and framework design patterns as supported models and workloads grow.
- Work directly with Enterprise customers 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; the posting also lists 3+ years as a baseline requirement.
- Significant open-source contributions; fluency in Swift, Kotlin and Python; or direct experience with Core ML, MLX Swift and LiteRT.
Tools
Python, Swift, Kotlin, Core ML, MLX Swift, LiteRT, ONNX, WebGPU
AI focus
Build and optimize on-device AI inference frameworks; productionize research prototypes and may contribute to AI research conference submissions.
Model-written summary · the original posting is authoritative