jobs

Senior Software Engineer, AI Runtime

P-1428 At 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 buildin

In inglese

Company

Databricks

Location

Mountain View, California; San Francisco, California

Status

Open

Posted

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

Databricks is hiring a senior software engineer for AI Runtime, its managed platform for large-scale GPU training and fine-tuning. The work spans the managed training stack: scheduling and capacity, distributed parallelism, fault tolerance, checkpointing and restore, data loading, and GPU utilization. It also covers APIs and tooling, mentoring, and work with product and research teams. The posting asks for five or more years on large-scale distributed systems, distributed training frameworks such as PyTorch, FSDP, DeepSpeed or Megatron, and GPU interconnect performance. Training runs span thousands of accelerators.