🛠 Tech Stack

💼 About This Role

You'll design and build large-scale experiment infrastructure for Databricks AI Research, working on distributed training and inference across thousands of GPUs. Your work enables researchers to iterate quickly and brings novel AI from prototype to production.

🎯 What You'll Do

• Design and implement infrastructure for large-scale experiments and model training.
• Build job submission, scheduling, and monitoring abstractions for researchers.
• Create tooling to improve research developer productivity and reduce iteration time.
• Influence the long-term roadmap for research computation at Databricks.

📋 Requirements

• BS/MS/PhD in Computer Science or related field
• 5+ years of software engineering experience with distributed systems or infrastructure.
• Deep experience with distributed systems, data pipelines, or large-scale backend services involving GPUs or cloud providers.
• Proficiency in systems programming languages (C++, Rust, Go, Java, Scala) to design complex services.

✨ Nice to Have

• Built or contributed to cluster schedulers (Kubernetes, Slurm, Ray, etc.).
• Understanding of ML training and inference workflows (distributed training, model parallelism, fine-tuning).
• Experience driving complex systems from prototype to stable service.

🎁 Benefits & Perks

• 💰 Equity and annual performance bonus included in total compensation.
• 🏖️ Comprehensive benefits and perks tailored to location.
• 🚀 Work on cutting-edge AI research with world-class scientists and engineers.
• 🌍 Inclusive culture with commitment to diversity and equal opportunity.

📨 Hiring Process

Estimated timeline: 2-4 weeks · AI estimate

1. 1Recruiter Screen· 30 min
2. 2Technical Phone Screen· 60 min
3. 3On-site Interviews (4-5 rounds)· 4 hours