"deepspeed" Jobs

6 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-LM. Every listing is re-checked daily and closed roles are removed.

Showing 6 of 6 results

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 1 month 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 3 weeks ago

DeepSpeed Composer Megatron-LM PyTorch NCCL

Flexible on stack Heavy meetings

Apply →

AI Research Scientist - Datadog AI Research (DAIR)

Datadog New York, New York, USA $140k–$400k/yr Published 1 year ago

multimodal learning Machine Learning DeepSpeed AI/ML reinforcement learning

Apply →

AI Research Scientist - Datadog AI Research (DAIR) Remote

Datadog Paris, France Published 1 year ago

Machine Learning DeepSpeed AI/ML reinforcement learning Megatron-LM

Apply →

Senior AI Compute Infrastructure Engineer Remote

Kraken United Kingdom Published 2 months ago

Ray TorchServe CUDA DeepSpeed vLLM

Apply →

Staff Research Engineer, Post-training & Evaluation Remote

Reddit Remote - United States $216.7k–$303.4k/yr Published 5 months ago

vLLM lm-eval-harness PyTorch Python MLflow

Apply →