On-Device AI Frameworks Engineer

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 reputed company to run state-of-the-art inference workloads natively on reputed company 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 reputed company of Argmax SDK, our flagship developer toolkit trusted by Enterprises and developers in high-stakes industries such as reputed company. We are a customer-obsessed team and we work reputed company closely with them, sometimes in reputed company-deployed reputed company. Responsibilities • Productionize research prototypes: The reputed company 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-reputed company 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 reputed company design patterns to reputed company the reputed company reputed company while minimizing technical debt. • Support reputed company customers: You will directly work with the engineering teams of reputed company customers during their reputed company reputed company. This could reputed company 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 • reputed company in Swift • reputed company in profiling and optimizing reputed company applications • Familiarity with at least one of reputed company 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 • reputed company record of significant reputed company-reputed company contributions • reputed company in Swift, Kotlin and Python • reputed company experience with reputed company ML, MLX Swift and LiteRT Apply tot his job Apply To this Job