Sr Worldwide Specialist Solutions Architect - GenAI, Training & Inference

Skills Required

Requires deep familiarity across the stack –

• Compute infrastructure (Amazon EC2, Lustre)
• ML frameworks (PyTorch, JAX)
• Orchestration layers (Kubernetes, Slurm)
• Parallel computing (NCCL, MPI)
• MLOps
• Target use cases in the cloud

You will drive the development of:

• Communication libraries (NCCL, MPI)
• Frameworks (PyTorch, NeMO, JAX)
• Models (Llama, Nemotron)

We also work with:

• Containers (Docker, Enroot)
• Orchestrators (EKS)
• Schedulers (Slurm)

Diverse Experiences

Amazon values diverse experiences. Even if you do not...