> Markdown version of [/jobs/ext/4694832-on-device-ai-frameworks-engineer-staff](https://www.wearedevelopers.com/jobs/ext/4694832-on-device-ai-frameworks-engineer-staff). Every page supports `.md` or `Accept: text/markdown`. Links point to the HTML versions so they work for humans too. Agent guide: [/agents.md](https://www.wearedevelopers.com/agents.md).

---

# On-device AI Frameworks Engineer (Staff)

- **Company:** Argmax, Inc.
- **Location:** Palo Alto, CA, United States
- **Experience:** Experienced
- **Contract:** Permanent contract
- **Skills:** Application Programming Interfaces (APIs), Artificial Intelligence, Android Software Development, Apple IOS, Apple Mac Systems, Application Frameworks, Profiling, Programming Tools, Python (Programming Language), Test-Driven Development (TDD), On-device Inference, Swift (Programming Language), Technical Debt, Kotlin, ONNX (Open Neural Network Exchange) Format, Free and Open-Source Software
- **Published:** November 12, 2026
- **Apply:** https://www.careerbuilder.com/job-details/on-device-ai-frameworks-engineer-staff-palo-alto-ca--c4b38800-35c9-46a6-80e4-9f73e515b4e0

## About the Role

Artificial Intelligence (AI), Android, Application Programming Interface (API), Research Skills, Artificial Intelligence (AI), Apple, Healthcare, Programming Tools, Prototyping, Blueprints, * 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

Preferred Qualifications

 * 5+ years of hands-on experience in SDK or Frameworks development for iOS or Android
 * Track record of significant open-source contributions
 * Fluency in Swift, Kotlin and Python
 * Direct experience with Core ML, MLX Swift and LiteRT

## Description

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.

AI Frameworks are at the core of Argmax SDK, our flagship developer toolkit trusted by Enterprises and developers in high-stakes industries such as healthcare. Argmax is a customer-obsessed team and we work very closely with them, sometimes in forward-deployed capacity.

Responsibilities

 * Productionize research prototypes: The Applied Research team will come to you with a Python prototype that provides the blueprints for a new feature (example) for one of our AI Frameworks. You will collaborate with them in turning this standalone prototype into a production-ready implementation in a test-driven development fashion. This collaboration may include experimentation and benchmarking that leads to top-tier AI research conference submissions (example).
 * Contribute to SDK and Frameworks design: As the number of supported models and workloads grow, you will see around corners and recommend scalable design patterns to absorb the code growth while minimizing technical debt.
 * Support Enterprise customers: You will directly work with the engineering teams of Enterprise customers during their onboarding journey. This could range from a Q&A session to customizing an Argmax feature to fit a particular customer requirement.