On-device AI Frameworks Engineer (Staff)

Argmax, Inc. is seeking a Staff Engineer to join their On-device AI Frameworks team. The role involves designing, implementing, and optimizing software frameworks for state-of-the-art inference workloads on Apple and Android devices, while collaborating 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