> Markdown version of [/jobs/ext/4003217-senior-software-engineer-ai-runtime](https://www.wearedevelopers.com/jobs/ext/4003217-senior-software-engineer-ai-runtime). Every page supports `.md` or `Accept: text/markdown`. Links point to the HTML versions so they work for humans too. Agent guide: [/agents.md](https://www.wearedevelopers.com/agents.md).

---

# Senior Software Engineer, AI Runtime

- **Company:** Databricks
- **Location:** Mountain View, CA, United States
- **Experience:** Expert
- **Salary:** $160,000.0 - $225,000.0
- **Contract:** Permanent contract
- **Skills:** Artificial Intelligence, Computer Programming, Extract Transform Load (ETL), Distributed Systems, Software Engineering, Graphics Processing Unit (GPU), High Performance Computing, DeepSpeed, Information Technology, Machine Learning Operations, Megatron, Databricks
- **Published:** October 29, 2026
- **Apply:** https://www.careerbuilder.com/job-details/senior-software-engineer-ai-runtime-mountain-view-ca--1241e658-767f-49e5-95f0-76d23d708439

## About the Role

Software Engineering, Artificial Intelligence (AI), Artificial Intelligence (AI), Drug Discovery, GPU (Graphics Processing Unit), Kernel Programming, Blog, Software Engineering, Problem Solving Skills, Vehicle Fleets, * 5+ years in large-scale distributed systems, with experience in GPU training, high-performance computing, or ML systems.

 * Experience training large models with FSDP, DeepSpeed, or Megatron, and familiarity with parallelism strategies.
 * Working knowledge of GPU performance, including accelerator architecture, high-speed interconnects, and kernels.
 * Strong understanding of training resilience patterns, such as checkpointing, failure detection, and automatic recovery.
 * A background in building and running multi-tenant cloud platforms as products.
 * A track record of owning complex features end to end and seeing them through to customer impact.
 * The clarity to explain technical designs in writing and in person, and a reputation as a teammate others learn from.
 * BS in Computer Science or a related field (MS or PhD preferred).

Pay Range Transparency

## Description

* Build core parts of AI Runtime and contribute to Databricks' technical direction in AI training infrastructure.
 * Develop reliability and observability features for a fleet that keeps growing in scale and complexity.
 * Improve AI Runtime's developer experience for people and agents, so even the largest training jobs feel as simple as a first experiment.
 * Help prepare AI Runtime for new GPU generations and workload types.
 * Take on hard multi-node training problems, from orchestration and GPU scheduling to data loading and checkpointing.
 * Make training faster and cheaper by raising efficiency across model architectures and GPU accelerators.
 * Work with research, product, and customers on what we build next and own features from design through production.
 * Champion engineering excellence, raise the bar, and help the engineers around you grow.