Senior Software Engineer, AI Runtime

Job summary

Responsibilities

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.

• 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.