On-device AI Frameworks Engineer (Intern)

Argmax • Palo Alto, CA

12h • Hybrid

About The Position

Argmax is seeking an intern to join its On-device AI Frameworks team. The role involves designing, implementing, and optimizing software frameworks that provide developer-friendly APIs for running state-of-the-art inference workloads natively on Apple and Android devices. The intern will collaborate with engineers and researchers on on-device inference technology and its adoption. AI Frameworks are a core part of Argmax SDK, a developer toolkit used by enterprises and developers in high-stakes industries such as healthcare. The team works closely with customers, sometimes in a forward-deployed capacity.

The intern will help implement new model support and features as the Argmax SDK grows, learning from senior engineers as they develop implementation specifications. The intern will also review customer issue reports, request information, reproduce and diagnose issues, and propose workarounds and permanent fixes, with senior engineers supporting their development toward handling this work autonomously over time.

The role offers three days per week in the office in Palo Alto, California, or Manhattan, New York. The Palo Alto office has on-site amenities, including chef-catered meals. If the internship is successful, there may be an opportunity to join Argmax, a fast-growing early-stage startup working to make on-device AI the default by bringing real-time AI workloads to the edge.

Requirements

• 3–6 months of hands-on experience in 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 Haves

• At least 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 ExecuTorch.

Responsibilities

• Contribute to SDK and frameworks implementation, including new model support and features.
• Learn from senior engineers as they design implementation specifications and work on the resulting implementation.
• Review customer issue reports and request supporting information.
• Reproduce and root-cause customer issues.
• Propose viable workarounds and permanent fixes for customer issues.

Benefits

• If the internship is successful, an opportunity to join Argmax.
• The Palo Alto office offers comprehensive on-site amenities, including chef-catered meals.
• Platinum-tier healthcare with 90% employer contribution, including dependents.

What This Job Offers

• Career Level: Intern
• Education Level: No Education Listed