Acceler8 Talent

Compensation

Full Description

We are looking for a Staff Engineer to join our growing On-device AI Frameworks team! 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. We are 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 in-house feature to fit a particular customer requirement. Qualifications • 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