About this role
Senior AI Infrastructure Engineer - Model Training at Kodiak. Location: Mountain View, California, United States. Role: designing pipelines, optimizing training, profiling systems Requirements: Degree in CS or related field,2+ years ML systems experience,expertise in distributed training,high-performance data pipelines,GPU performance and profiling,Python and PyTorch skills. Category: Engineering Seniority: Entry Level Tools: PyTorch, PyTorch DDP/FSDP, DeepSpeed, Megatron, NCCL, WebDataset, MosaicML Streaming, MDS, Nsight, PyTorch Profiler, Python, C++, CUDA, Triton, NVLink, InfiniBand Commitment: Full Time Workplace: Onsite Languages: English