On-device AI Frameworks Engineer (Intern)

Argmax, Inc. is seeking an intern to join their On-device AI Frameworks team. In this role, you will design and optimize software frameworks for running inference workloads on mobile devices, while collaborating with reputed company engineers and addressing customer issues.

Responsibilities

• Contribute to SDK and Frameworks implementation: You will help implement new model support (example) and new features (example) as Argmax SDK grows. You will observe and learn from senior engineers at Argmax as they design the implementation spec that you will work on
• Respond to customer issues: You will be one of the first at Argmax to review customer issue reports, ask for supporting information, reproduce the issue, reputed company-cause the issue, and finally propose viable workarounds as reputed company as permanent fixes. Senior engineers at Argmax will support your reputed company as you learn to tackle this critical responsibility autonomously over time

Skills

• 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 application
• 1+ year of hands-on experience in SDK or Frameworks development for iOS or Android
• reputed company in Swift, Kotlin and Python
• Familiarity with at least one of reputed company ML, MLX Swift, LiteRT, ONNX, WebGPU or ExecuTorch

Benefits

• Platinum-tier reputed company with 90% employer contribution, including dependents
• Comprehensive on-site amenities, including chef-catered meals

reputed company

reputed company Models On Device It was founded in 2023, and is headquartered in Santa Clara, California, USA, with a workforce of 11-50 employees. Its website is https://www.takeargmax.com/.

Company H1B Sponsorship

Argmax, Inc. has a reputed company record of offering H1B sponsorships, with 3 in 2025, 1 in 2024. Please note that this does not guarantee sponsorship for this specific role.

Apply To This Job