Technical Lead, Machine Learning
A1 · Palo Alto, CA · 2 wk ago
HybridEngineeringFull-time
What You'll Do
- Own end-to-end ML system execution: data pipelines, training workflows, evaluation systems, inference architecture, and deployment.
- Fine-tune and adapt models using state-of-the-art 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, in partnership with research leadership.
- Own production deployment, including GPU optimization, memory efficiency, latency reduction, and scaling policies.
- Collaborate closely with application engineering to integrate ML systems cleanly into backend, mobile, and desktop products.
- Make pragmatic trade-offs and ship improvements quickly, learning from real usage.
- Work under real production constraints: latency, cost, reliability, and safety.
Outcomes
- Research and models reliably translate into production-ready solutions with clear performance and quality targets.
- ML pipelines, training loops, and inference systems are stable, efficient, and maintainable.
- Production issues are detected, debugged, and resolved quickly, minimizing user impact.
- Team members are supported, aligned, and able to deliver high-impact ML work with minimal friction.
- Iterations on models and systems are measurable, safe, and improve user experience over time.
Tech Stack
Python
PyTorch / JAX
GPU-based training and inference system
Ideal Experience
- You have built or shipped real ML systems used by people, not just demos.
- You are comfortable working with large models and understanding their failure modes.
- You write strong, production-grade code and care about system correctness.
- You are self-directed, pragmatic, and take full ownership of outcomes.
- You communicate clearly and collaborate well in small, high-trust teams.
How We Work
The best products today in the world were built by small, world class teams. We are a high talent density and hands-on team. We make decisions collectively, move at rapid speed, striking a balance between shipping high quality work and learning.
Joining Our Team
Requires the ability to bring structure, exercise judgment, and execute independently.
Interview Process
- If there appears to be a fit, we'll reach out to schedule 3, but no more than 4 interviews.
- Applications are evaluated by our technical team members.
- Interviews will be conducted via virtual meetings and/or onsite.
- We value transparency and efficiency, so expect a prompt decision.
- If you've demonstrated the exceptional skills and mindset we're looking for, we'll extend an offer to join us. This isn't just a job offer; it's an invitation to be part of a team that's bringing AI to have practical benefits to billions globally.