LLM Pre-training & Distributed Engineer (AI Infrastructure)

Added
25 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, or Megatron-LM.
  • Optimize networking (InfiniBand/RDMA) and memory management to prevent OOM.
  • Automate checkpointing and failure recovery during long training runs.

๐ŸŽฏ Requirements

  • Deep expertise in 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 โ†’