Senior Software Engineer, AI RuntimeActive$160K
The opportunity
At Databricks, we run the world's largest AI platform. We believe the next breakthroughs will come from models trained on domain-specific enterprise data, not from bigger general-purpose models alone.
What you'll do
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.
What they're looking for
- 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.
- + 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.
