Todas as vagas
ML Infrastructure Engineer
Nebius
Prague +1 · Aceita remotoSênior (5–10 anos)Tempo integralPublicada há 15 dias
Sobre a vaga
O que você vai fazer
Lead and support the benchmarking of GPU platforms to evaluate performance for machine learning and AI workloads. This includes profiling GPU performance at the system and kernel level and optimizing ML workloads to resolve bottlenecks.
O que eles procuram
Requires a profound understanding of machine learning theoretical foundations and deep experience with modern frameworks like PyTorch and JAX. Candidates must have strong knowledge of the GPU stack, including CUDA and NCCL, and familiarity with containerized environments.
Descrição completa