Staff Machine Learning Engineer

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 latency reduction
• Collaborate with application engineering to integrate ML systems into backend, mobile, and desktop products

Requirements

• Experience building or shipping real ML systems used by people
• Proficiency in writing strong, production-grade code
• Comfort working with large models and understanding their failure modes
• Ability to work under production constraints like latency, cost, and reliability
• Strong communication skills and ability to work in small, high-trust teams

Preferred Qualifications

• Experience with Python, PyTorch, or JAX
• Experience with GPU-based training and inference systems