Staff Software Engineer, AI Runtime

Job summary

Responsibilities

As a Staff Software Engineer, you'll work on the systems behind all of this, solving hard technical problems and shaping major parts of AI Runtime.

• Drive major parts of AI Runtime's architecture and help shape Databricks' long-term technical direction in AI training infrastructure.
• Design the reliability and observability systems for a fleet that keeps growing in scale and complexity.
• Lead improvements to AI Runtime's developer experience for people and agents, so even the largest training jobs feel as simple as a first experiment.
• Help choose the technical bets that prepare AI Runtime for new GPU generations and workload types.
• Take on the hardest multi-node training problems yourself, from orchestration and GPU scheduling to data loading and checkpointing.