Staff On-Device AI Frameworks Engineer (iOS/macOS)

Argmax, Inc.; Palo Alto (CA); Hybrid; USD 130,000 - 160,000; Full time

Benefits offered by this job

Top-of-market equity

Performance-based equity refreshers

Comprehensive on-site amenities

Remote work possible

Platinum-tier healthcare

401(k) match

Quarterly team-building weeks

Job summary

Argmax, Inc. in Palo Alto, CA is looking for a Staff Engineer for its On-device AI Frameworks team. This role focuses on designing and optimizing software frameworks for native inference workloads on both Apple and Android devices. You will work closely with research teams to transform prototypes into production-ready solutions and directly support enterprise customers during implementation. The company offers competitive equity, top-tier healthcare, and in-office perks in a collaborative environment.

Qualifications

• 3+ years of hands-on experience in SDK or Frameworks development for iOS or macOS.
• Fluency in Swift and experience with optimizing native applications.
• Familiarity with at least one of Core ML, MLX Swift, LiteRT, ONNX, WebGPU or ExecuTorch.

Responsibilities

• Productionize research prototypes into production-ready implementations.
• Contribute to SDK and Framework design for scalability.
• Support Enterprise customers throughout their onboarding journey.

Skills

SDK development

Frameworks development

Fluency in Swift

Optimization of native applications

Core ML familiarity

MLX Swift familiarity

LiteRT familiarity

Kotlin fluency

Python fluency

Job description

Argmax, Inc. in Palo Alto, CA is looking for a Staff Engineer for its On-device AI Frameworks team. This role focuses on designing and optimizing software frameworks for native inference workloads on both Apple and Android devices. You will work closely with research teams to transform prototypes into production-ready solutions and directly support enterprise customers during implementation. The company offers competitive equity, top-tier healthcare, and in-office perks in a collaborative environment.