On-device AI Frameworks Engineer (Staff)

Responsibilities

• Productionize research prototypes: Collaborate with the Applied Research team to turn standalone Python prototypes into production-ready implementations.
• Contribute to SDK and Frameworks design: Recommend scalable design patterns to handle increasing model support and workload diversity.
• Support Enterprise customers: Work directly with enterprise engineering teams to customize and integrate Argmax features according to specific customer requirements.

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

About Argmax

The infrastructure supporting AI applications is failing to keep up with user adoption. Argmax is addressing this gap by building the critical infrastructure needed to move real-time AI workloads to the edge. Our mission is to provide autoscaling, private, and compliant AI solutions without sacrificing accuracy.