jobs

Staff Software Engineer, AI Runtime

P-1930At Databricks, we are passionate about enabling data teams to solve the world's toughest problems — from making the next mode of transportation a reality to accelerating the development of medical breakthroughs. We do this by building

Company

Databricks

Location

Mountain View, California; San Francisco, California

Status

Open

Posted

2026-06-08T21:32:45+00:00

Databricks is hiring a staff software engineer for AI Runtime, its managed platform for large-scale GPU training and fine-tuning. The role covers architecture and scaling of the training stack: scheduling and capacity, multi-node orchestration, distributed parallelism, fault tolerance, checkpointing, and job launch experience. Applicants need 10+ years with large distributed systems and hands-on work in frameworks such as PyTorch, FSDP, DeepSpeed, or Megatron, plus GPU performance and interconnect knowledge. The posting highlights thousands-of-GPU fleets, long-running jobs, and cross-team technical leadership.