← All roles

Senior Software Engineer, AI Runtime

Permanentonsite5+ yrs📍 Mountain View, California; San Francisco, California, 🇺🇸 United States🗣 English

Build and scale large-scale GPU training infrastructure for AI models at Databricks.

You'll build and scale large-scale GPU training infrastructure for AI models using PyTorch, FSDP, DeepSpeed, and Megatron. The role involves working with GPU and high-speed networking technologies like NVLink and InfiniBand. This is a permanent, full-time senior position based onsite in Mountain View or San Francisco, California.

Tech stack
PyTorchFSDPDeepSpeedMegatronGPUNVLinkInfiniBand
Rate
Not stated by the agency
via

Membership is €29/month, cancel anytime: every rate, every original listing link, and a daily alert for roles matching your filters.

Found at a specialist agency · listed 14 August 2026 · InsideJobs links you to the original posting.