"deepspeed" Jobs

4 open tech roles matching “deepspeed”, taken straight from company career pages — not reposted from other job boards. Most in demand right now: DeepSpeed, PyTorch, Megatron. Every listing is re-checked daily and closed roles are removed.

Showing 4 of 4 results

Senior AI Infrastructure Engineer - Model Training Visa

Join Kodiak Robotics as a Senior AI Infrastructure Engineer to optimize model training for autonomous technology.

Kodiak Robotics Mountain View, CA $190k–$260k/yr Published 2 months ago

Megatron CUDA DeepSpeed PyTorch Python

Flexible on stack

Apply →

Senior Software Engineer, AI Runtime

Join Databricks as a Senior Software Engineer to build and scale a managed GPU training platform for AI models.

Databricks Mountain View, California; San Francisco, California $160k–$225k/yr Published 3 months ago

NVLink Megatron DeepSpeed PyTorch GPU

Flexible on stack 70% coding

Apply →

Sr. Engineering Manager, AI Runtime

Lead a high-performing engineering team at Databricks to enhance the AI Runtime product for GPU training infrastructure.

Databricks Mountain View, California; San Francisco, California $228.6k–$297.1k/yr Published 2 months ago

DeepSpeed Composer Megatron-LM PyTorch NCCL

Flexible on stack Heavy meetings

Apply →

Staff Software Engineer, AI Runtime

Join Databricks as a Staff Software Engineer to drive the architecture of a managed GPU training platform for large-scale AI models.

Databricks Mountain View, California; San Francisco, California $190k–$265k/yr Published 3 months ago

NVLink Megatron DeepSpeed PyTorch GPU

Apply →