On-device AI Frameworks Engineer (Staff)

Argmax, Inc. is a company focused on building critical infrastructure for real-time AI workloads at the edge. They are seeking a Staff Engineer to join their On-device AI Frameworks team, where the primary responsibilities include designing and optimizing software frameworks for inference workloads on mobile devices, collaborating with research teams, and supporting enterprise customers during onboarding.

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