About the job
About the Role
A1 is at the forefront of developing an advanced AI system designed to comprehend context across conversations, formulate strategic actions, and maintain continuity over time.
Your role will involve transforming innovative research into operational, production-grade machine learning systems. You will lead the execution layer of A1’s intelligence, overseeing training pipelines, inference systems, evaluation tools, and deployment processes.
What You'll Be Responsible For
Develop and manage comprehensive ML pipelines that encompass data management, training, evaluation, inference, and deployment.
Optimize and refine models utilizing cutting-edge methodologies such as LoRA, QLoRA, SFT, DPO, and model distillation.
Design and maintain scalable inference systems, ensuring a balance between latency, cost-effectiveness, and reliability.
Create and manage data systems to ensure access to high-quality synthetic and real-world training data.
Establish evaluation pipelines that assess performance, robustness, safety, and bias, collaborating closely with research leadership.
Oversee production deployment, focusing on GPU optimization, memory efficiency, latency reduction, and scaling strategies.
Engage closely with application engineering teams to seamlessly integrate ML systems into backend, mobile, and desktop applications.
Make informed trade-offs and rapidly implement improvements based on real-world usage.
Operate under real production constraints including latency, cost, reliability, and safety considerations.
Technical Expertise
Proficient in Python
Experience with PyTorch and JAX
Familiarity with GPU-based training and inference systems
Preferred Qualifications
You have successfully built or deployed machine learning systems that deliver real value to users.
You possess a strong understanding of large models and their potential failure modes.
Your coding skills are robust, with a focus on production-grade quality and system integrity.
Self-motivated and pragmatic, you take full ownership of your contributions and outcomes.
You communicate effectively and excel in collaborative, high-trust team environments.
Our Working Environment
Our organization is structured to foster innovation and collaboration among talented individuals, committed to pushing the boundaries of AI technology.

