Senior Machine Learning Infrastructure Engineer
Core
Design and operate distributed training infrastructure for neural operator architectures on NVIDIA DGX B200 systems.
Role type
Senior Machine Learning Infrastructure Engineer
Builds
Distributed training pipelines, experiment tracking systems, model-serving infrastructure, and observability tools for Large Physics Models.
Domain
High-performance computing (HPC), GPU clusters, and large-scale ML training.
Deliverable
production ML models
Required skills
Distributed training (NCCL, FSDP, DDP, pipeline parallelism), Linux systems administration, Kubernetes, SLURM, Python, PyTorch, GPU cluster orchestration, performance optimization, data I/O optimization.
Preferred skills
Geometric deep learning, neural operators, mesh/point-cloud/graph architectures, HPC simulation workflows, model serving, experiment tracking, cloud GPU infrastructure (CoreWeave).
Technologies
NVIDIA DGX B200, NCCL, InfiniBand, NVLink, Kubernetes, SLURM, PyTorch, Python, CoreWeave.
