🛠 Tech Stack

💼 About This Role

You'll own the execution layer of A1's proactive AI chat app, building reliable production-grade ML systems for multi-step reasoning and real-world task completion. You'll translate research into scalable models, data pipelines, and inference systems under latency and cost constraints. This role sits at the intersection of research, infrastructure, and product, driving ML from training to deployment.

🎯 What You'll Do

• Own end-to-end ML system execution including data pipelines and deployment
• Fine-tune models using LoRA, QLoRA, SFT, DPO, and distillation
• Architect scalable inference systems balancing latency, cost, and reliability
• Design evaluation pipelines covering performance, robustness, and safety

📋 Requirements

• 5+ years experience building and shipping production ML systems
• Proficiency in Python and PyTorch or JAX
• Experience with GPU optimization and large model deployment
• Strong understanding of ML pipeline and system correctness

✨ Nice to Have

• Experience with synthetic data generation and data quality
• Background in conversational AI or task-oriented systems

🎁 Benefits & Perks

• 🏖️ Flexible PTO
• 🏥 Health insurance
• 💻 Remote-friendly hybrid
• 📈 Equity options
• 📚 Learning budget

📨 Hiring Process

3-4 interviews including technical assessments, conducted virtually or onsite.