Technical Lead, Machine Learning

About the Company

A1 is building a proactive smart assistant for everyday users to bring intelligence to conversations, errands, organizing, and workflows. The mission is to create an AI-native experience that handles multi-step reasoning and real-world task completion with high reliability.

Responsibilities

• Own end-to-end ML system execution including data pipelines, training workflows, evaluation systems, and inference architecture
• Fine-tune and adapt models using methods such as LoRA, QLoRA, SFT, DPO, and distillation
• Architect and operate scalable inference systems balancing latency, cost, and reliability
• Design and maintain data systems for high-quality synthetic and real-world training data
• Implement evaluation pipelines covering performance, robustness, safety, and bias
• Own production deployment including GPU optimization, memory efficiency, and scaling policies
• Collaborate with application engineering to integrate ML systems into backend, mobile, and desktop products

Requirements

• Experience building or shipping real ML systems used in production
• Proficiency in Python
• Experience with PyTorch or JAX
• Experience with GPU-based training and inference systems
• Ability to work with large models and understand their failure modes
• Strong production-grade coding skills
• Ability to work independently in a high-talent density team