"deepspeed" Jobs

9 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 9 of 9 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 4 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 5 months ago

NVLink Megatron DeepSpeed PyTorch GPU

Flexible on stack 70% coding

Apply →

Senior AI Research Engineer - GPU Systems & Kernels Hybrid

Join Datadog as a Senior AI Research Engineer to optimize GPU systems and kernels for large-scale model training and inference.

Datadog Paris, France Published 1 day ago

C++ Triton NVIDIA CUDA DeepSpeed

Flexible on stack

Apply →

Senior AI Research Engineer - LLM Training & Inference Infrastructure Hybrid

Join Datadog as a Senior AI Research Engineer to build large-scale LLM training and inference infrastructure.

Datadog Paris, France Published 1 day ago

C++ Ray SGLang DeepSpeed vLLM

Flexible on stack

Apply →

Software Engineer - Training Product

Join Baseten as a senior software engineer to develop cutting-edge AI training products and enhance user workflows.

baseten San Francisco Published 9 months ago

AI/ML PyTorch Kubernetes

Flexible on stack

Apply →

AI Research Scientist - Datadog AI Research (DAIR)

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

multimodal learning Machine Learning DeepSpeed AI/ML reinforcement learning

Apply →

Software Engineer - Training Infrastructure

Join Baseten as a Software Engineer to architect and develop scalable infrastructure for ML training platforms.

baseten San Francisco Published 1 year ago

Axolotl MLOps AWS HF Trainer Megatron

Flexible on stack

Apply →

AI Research Scientist - Datadog AI Research (DAIR) Hybrid

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 6 months ago

Ray TorchServe CUDA DeepSpeed vLLM

Apply →