Senior Software Engineer, AI Runtime
Databricks · Mountain View, California; San Francisco, California · Engineering · listed June 8, 2026
The shape of it
Seniority
Senior
Experience asked
5+ years
Where
On site
Stated pay
$160,000 – $225,000 USD
Requirements listed
8
Length
814 words
In the posting’s own words
As a Senior Software Engineer, you'll work on the systems behind all of this, solving hard technical problems and helping shape how AI Runtime evolves.
What it asks for · 8
- 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).
What the job covers
- 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.
Degree language
- BS in Computer Science or a related field (MS or PhD preferred).
Tools and skills named
Models & research
- GPU7×
- Fine-tuning
- Machine learning
Cloud & infra
- Distributed systems
- Observability
- Serverless
Data
- Databricks3×
Frameworks
- Node.js
Words the posting leans on
- training10×
- models7×
- gpu6×
- runtime5×
- build3×
- customers3×
- experience3×
- features3×
- next3×
- product3×
- systems3×
- technical3×
- accelerator2×
- agents2×
- already2×
- architecture2×
Counted from the posting after the mission statement and the legal notices are set aside. The ones near the top are the ones a screener is looking for.
The posting, your resume, and the gaps between them. One click loads all three.