ML Systems Technical Lead — End-to-End AI Deployment

A1 is seeking a Technical Lead for Machine Learning to take ownership of the execution layer of their AI smart assistant.

In this role, you will translate research into reliable, production-grade ML systems, ensuring models are trainable and deployable. Responsibilities include managing end-to-end ML processes, optimizing performance, and collaborating closely with engineering teams to deliver innovative solutions.

The ideal candidate will have experience in building ML systems and a strong coding background in Python, alongside tools like PyTorch and JAX.