Machine Learning Performance Engineer

What this role involves

This role focuses on optimizing the performance of machine learning models for both training and inference. The engineer will work with low-level GPU and systems programming, using a whole-systems approach to improve efficiency. It is part of Jane Street's growing ML team, which supports the firm's trading operations.

Skills this posting asks for

• ptx
• sass
• cuda
• triton
• cutlass
• cub
• thrust
• cudnn
• cublas
• cuda gdb
• nsight systems
• nsight compute
• infiniband
• roce
• gpudirect
• pxn
• nvlink
• nccl
• mpi
• cuda graph launch
• tensor core arithmetic
• warp-level synchronization
• asynchronous memory loads
• ml techniques

From the employer’s posting

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