"megatron" Jobs

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

Showing 4 of 4 results

Member of Technical Staff - Pre-Training Infra Visa

Build and scale distributed training systems for frontier model pre-training at Reflection AI.

Reflection AI San Francisco, CA Published 5 months ago

Megatron DeepSpeed distributed training NCCL

Flexible on stack

Apply →

Member of Technical Staff - Research Software Engineer Visa

Join Reflection AI as a Research Software Engineer to bridge research and production in cutting-edge AI training systems.

Reflection AI New York, NY Published 6 months ago

Ray JAX Slurm RDMA PyTorch

Flexible on stack

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 →

Staff Applied ML Engineer

Join CoreWeave as a Staff Applied ML Engineer to tackle challenges in continuous learning for AI agents with a highly autonomous team.

Coreweave Sunnyvale, CA / Bellevue, WA $185k–$275k/yr Published 5 months ago

Megatron Temporal Large Language Models CUDA Machine Learning

Flexible on stack 60% coding

Apply →