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 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
Apply tot his job
Apply To this Job
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
Apply tot his job
Apply To this Job