Software Engineer

At Databricks, we are obsessed with enabling data teams to solve the world’s toughest problems. We build and run the world’s best data and AI platform.

• Develop and run the research stack that powers Databricks AI Research.
• Design and build services that schedule, orchestrate, and observe large‑scale training and inference experiment workloads across thousands of GPUs.
• Improve dev tooling and ensure researchers can iterate quickly without sacrificing reliability, efficiency, or security.
• Partner closely with research scientists, ML engineers, and platform teams.
• Design and implement infrastructure for large‑scale experiments, data processing, and model training.
• Enable researchers to go from idea to large‑scale experiment in minutes.
• Create tooling that improves research developer productivity.
• Influence the long‑term roadmap for research computation.
• Serve as a technical mentor.

• BS/MS or PhD in Computer Science or related field.
• 5+ years of software engineering experience, including substantial time working on large‑scale distributed systems or infrastructure.
• Deep experience with building and operating distributed systems, data pipelines, or large‑scale backend services, ideally involving GPUs, clusters, or major cloud providers.
• Proficient in one or more systems programming languages (e.g., C++, Rust, Go, Java, Scala).
• Built or significantly contributed to cluster schedulers, resource managers, or large‑scale job orchestration systems (e.g., Kubernetes, Slurm, Ray, custom internal systems).
• Understand modern ML training and inference workflows.
• Can move fast and be pragmatic while caring about operational excellence.
• Communicate clearly with both researchers and engineers.

• Comprehensive health coverage including medical, dental, and vision.
• 401(k) Plan.
• Equity awards.
• Flexible time off.
• Paid parental leave.
• Family Planning.
• Gym reimbursement.
• Annual personal development fund.
• Work headphones reimbursement.
• Employee Assistance Program (EAP).
• Business travel accident insurance.
• Mental wellness resources.