Machine Learning Performance Engineer

What this role involves

We are looking for an engineer with experience in low-level systems programming and optimization to join our growing ML team. The role focuses on optimizing model performance for training and inference, involving CUDA, GPU knowledge, and networking technologies. The ideal candidate has deep systems knowledge and experience with modern ML tools.

Skills this posting asks for

• cuda
• ptx
• sass
• warps
• cooperative groups
• tensor cores
• memory hierarchy
• cuda gdb
• nsight systems
• nsight compute
• triton
• cutlass
• cub
• thrust
• cudnn
• cublas
• infiniband
• roce
• gpudirect
• pxn
• rail optimization
• nvidia nvlink
• nccl
• mpi

From the employer’s posting

We are looking for an engineer with experience in low-level systems programming and optimization to join our growing ML team.  Machine learning is a critical pillar of Jane Street's global business. Our ever-ev…