Argmax
On-device AI Frameworks Engineer (Intern)
Palo Alto, CA +1 · Internship · Posted Nov 23, 2025
- Location
- Palo Alto, CA
Palo Alto, California, United States - Experience
- 3–6 months of hands-on SDK or frameworks development for iOS or Android; preferred: 1+ year.
About the job
Develop and optimize SDK frameworks that enable state-of-the-art AI inference natively on Apple and Android devices, while helping investigate and resolve customer issues.
What you'll do
- Implement new model support and features in the SDK and frameworks.
- Review customer issue reports, gather information, reproduce and diagnose issues, and propose workarounds or permanent fixes.
What they ask for
- Hands-on SDK or frameworks development for iOS or Android.
- Familiarity with Swift, or with both Kotlin and C++.
- Familiarity with profiling and optimizing mobile applications.
Nice to have
- At least 1 year of hands-on SDK or frameworks development for iOS or Android.
- Fluency in Swift, Kotlin, and Python; familiarity with at least one of Core ML, MLX Swift, LiteRT, ONNX, WebGPU, or ExecuTorch.
Tools
Swift, Kotlin, C++, Python, Core ML, MLX Swift, LiteRT, ONNX
AI focus
Build frameworks for native on-device AI inference and expand model support.
Model-written summary · the original posting is authoritative