Senior Staff Software Engineer, AI Runtime

Job summary

Responsibilities

As a Senior Staff Software Engineer, you'll work on the systems behind all of this, solving the hardest technical problems and shaping where AI Runtime goes next.

• Drive AI Runtime's architecture and lead Databricks' long-term technical direction in AI training infrastructure.
• Define the reliability and observability strategy for a fleet that keeps growing in scale and complexity.
• Shape AI Runtime's developer experience for people and agents, so even the largest training jobs feel as simple as a first experiment.
• Anticipate industry trends and 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.