"megatron lm" Jobs

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

Showing 5 of 5 results

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 →

Senior AI Compute Infrastructure Engineer Remote

Kraken United Kingdom Published 2 months ago

Ray TorchServe CUDA DeepSpeed vLLM

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 Engineer - Datadog AI Research (DAIR) Remote

Datadog Paris, France Published 11 months ago

C++ Ray JAX CUDA DeepSpeed

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 →