Device AI Frameworks Engineer; Intern

Overview

We are looking for an intern to join our On-device AI Frameworks team at Argmax! In this role, you will design, implement and optimize software frameworks that expose developer-friendly APIs to run state‑of‑the‑art inference workloads natively on Apple and Android devices. You will collaborate closely with industry leaders, advancing the frontiers of on-device inference technology.

• Contribute to SDK and Frameworks implementation: help implement new model support and features as Argmax SDK grows.
• Respond to customer issues: review issue reports, reproduce problems, root‑cause them, and propose workarounds or fixes.

• 3-6 months of hands‑on experience in SDK or Frameworks development for iOS or Android.
• Familiarity with Swift OR (Kotlin AND C++).
• Familiarity with profiling and optimizing mobile applications.

• 1+ year of hands‑on experience in 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 Execu Torch.
• If internship is successful, opportunity to join Argmax.
• 3 days a week in the office from Palo Alto, CA or Manhattan, NY.
• Palo Alto office offers comprehensive on‑site amenities.
• Platinum‑tier healthcare with 90% employer contribution, including dependents.

AI applications are scaling in user adoption at unprecedented rates. The infrastructure is crumbling, and we are building critical infrastructure to bring real‑time AI workloads to the edge. Our mission is to make on‑device the default way to build AI applications.