Argmax, Inc. is looking for a Staff Engineer to join their On-device AI Frameworks team. In this role, you will design and optimize software frameworks for state-of-the-art inference workloads on ...
Argmax
1 job near Columbus, OH
Argmax, Inc. is looking for a Staff Engineer to join their On-device AI Frameworks team. In this role, you will design and optimize software frameworks for state-of-the-art inference workloads on ...
Full-time
Re-posted 21 days ago
Job description
Job Summary:
Argmax, Inc. is looking for a Staff Engineer to join their On-device AI Frameworks team. In this role, you will design and optimize software frameworks for state-of-the-art inference workloads on mobile devices, collaborating closely with industry-leading engineers and researchers.
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.
Qualifications:
Required:
• 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:
• 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
Company:
Foundation Models On Device Founded in 2023, the company is headquartered in Santa Clara, USA, with a team of 11-50 employees. The company is currently Early Stage.
Argmax, Inc. is looking for a Staff Engineer to join their On-device AI Frameworks team. In this role, you will design and optimize software frameworks for state-of-the-art inference workloads on mobile devices, collaborating closely with industry-leading engineers and researchers.
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.
Qualifications:
Required:
• 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:
• 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
Company:
Foundation Models On Device Founded in 2023, the company is headquartered in Santa Clara, USA, with a team of 11-50 employees. The company is currently Early Stage.