Technical Lead, Machine Learning

About the Role

A1 is building a proactive AI chat app for everyday users to bring intelligence to conversations, errands, organizing and workflows. Unlike traditional chat-based applications, our product focuses on achieving high reliability for long-running workflows, persistent context, and real-world task completion. The system must handle multi-step reasoning, interact with external tools, and remain reliable despite non-deterministic model behavior.

As Technical Lead, Machine Learning, you own the execution layer of A1’s intelligence. You translate research direction into reliable, scalable, production-grade ML systems. This role sits at the intersection of research, infrastructure, and product. You are responsible for making models trainable, deployable, observable, and performant under real-world constraints.

What You’ll Do

• Own end-to-end ML system execution: data pipelines, training workflows, evaluation systems, inference archite...