Senior Member of Technical Staff, Machine Learning

About the Role

A1 is building a proactive AI system that carries work forward across conversations, tools, and time.

You will be a core owner of significant ML systems in production. This role is for engineers who can independently drive complex ML problems from idea to deployment and make strong trade-offs along the way.

Focus

• Build core ML systems that power a proactive, long-horizon AI product.

• Own work end-to-end: data preparation, training, evaluation, inference, and iteration.

• Turn research ideas into working systems that run reliably in production.

• Debug model failures and system issues using real production signals.

• Iterate quickly: ship, measure outcomes, refine, and repeat.

• Collaborate closely with research, product, and engineering to deliver real user impact.

• Mentor and review work from other ML engineers through example and technical judgment.

• Work under real production constraints: latency, cost, reliability, and safety

Tech Stack

• Python

• PyTorch / JAX

• GPU-based training and inference systems

Ideal Experience

• You have built and shipped ML systems used by real users.

• You understand how modern ML models behave — and misbehave — in production.

• You write strong, production-quality code and think in systems, not scripts.

• You take ownership, work independently, and push work across the finish line.

• You learn fast, communicate clearly, and improve through iteration.

How We Work

Our organization is very flat and our team is small, highly motivated, and focused on engineering and product excellence. All members are expected to be hands-on and to contribute directly to the company’s mission.

Interview process

If there appears to be a fit, we'll reach to schedule 3, but no more than 4 interviews.

Applications are evaluated by our technical team members. Interviews will be conducted via virtual meetings and/or onsite.

We value transparency and efficiency, so expect a prompt decision. If you've demonstrated the exceptional skills and mindset we're looking for, we'll extend an offer to join us. This isn't just a job offer; it's an invitation to be part of a team that's bringing AI to have practical benefits to billions globally.