# On-device AI Frameworks Engineer (Staff) at Argmax

- Company: Argmax
- What the company does: Backed by General Catalyst.
- Company website: https://argmaxml.com
- Type: Startups (AI role)
- Level: Senior
- Location: Palo Alto, CA
- Work setup: On-site
- Posted: 2025-10-29
- Apply by: 2026-10-08
- Apply: https://job-boards.greenhouse.io/argmax/jobs/4052845009
- Page: https://www.1752.vc/careers/jobs/argmax-on-device-ai-frameworks-engineer-staff/

## About the role

We are looking for a Staff Engineer to join our growing 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-leader engineer and researcher colleagues, advancing the frontiers of on-device inference technology and accelerating its market adoption.

## What they're looking for

- 3+ years of hands-on experience in 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

Tags: SDK