"megatron lm" Jobs

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

Showing 6 of 6 results

ML Systems Engineer Visa

Join Periodic Labs as an ML Systems Engineer to build and optimize large-scale training and reinforcement learning infrastructure.

Periodic Labs Menlo Park, CA $250k–$350k/yr Published 4 months ago

Ray TensorRT-LLM Triton SGLang CUDA

Flexible on stack

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 →

Senior AI Compute Infrastructure Engineer Remote

Kraken United Kingdom Published 4 months ago

Ray TorchServe CUDA DeepSpeed vLLM

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 →

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 Software Engineer - Perf and Benchmarking

Join CoreWeave as a Senior Engineer to enhance Kubernetes-native benchmarking services and improve performance across a global infrastructure.

Coreweave Sunnyvale, CA / Bellevue, WA $182k–$242k/yr Published 4 months ago

PCIe NVLink C++ Grafana CUDA

Flexible on stack

Apply →