Member of Technical Staff, 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 transform non-AI-native applications into intelligent systems capable of multi-step reasoning and real-world task completion.

Responsibilities

• Build and improve ML components across data, training, evaluation, and inference
• Fine-tune and adapt models as part of larger production systems
• Implement evaluation and testing to understand model behavior
• Build and maintain data pipelines for real-world and synthetic data
• Debug model issues, performance problems, and production incidents
• Ship improvements iteratively under production constraints like latency, cost, and reliability

Requirements

• Strong foundations in machine learning and modern neural architectures
• Hands-on experience training, fine-tuning, or deploying ML models
• Ability to write production-quality code
• Proficiency with Python and PyTorch or JAX
• Experience with production ML systems running on GPUs

Preferred Qualifications

• Ability to work through ambiguity with guidance
• Bias toward shipping, iteration, and continuous improvement
• Eagerness to learn from real-world production systems