LLM Pre-training  Distributed Engineer (AI Infrastructure)

Added
24 minutes ago
Type
Full time
Salary
Salary not provided

Related skills

python kubernetes pytorch deepspeed cuda

πŸ“‹ Description

  • Orchestrate distributed training across 1,000+ GPUs using PyTorch/DeepSpeed/Megatron-LM.
  • Tune InfiniBand/RDMA networking and memory to prevent out-of-memory errors.
  • Automate checkpointing and failure recovery during month-long training runs.

🎯 Requirements

  • 3D parallelism (Data, Tensor, Pipeline).
  • Experience managing SLURM or Kubernetes-based GPU clusters.
  • Strong systems engineering background (C++, CUDA, Python).
Share job

Meet JobCopilot: Your Personal AI Job Hunter

Automatically Apply to Engineering Jobs. Just set your preferences and Job Copilot will do the rest β€” finding, filtering, and applying while you focus on what matters.

Related Engineering Jobs

See more Engineering jobs β†’