About this role

Jane Street is seeking a Machine Learning Performance Engineer to optimize the performance of ML models for training and inference. This role involves a deep dive into low-level systems programming, GPU optimization, and whole-system performance analysis within a fast-paced trading environment.

What we are looking for

6

Optimize ML model performance for training and inference
Low-level systems programming and optimization
Whole-systems approach including storage, networking, and GPU considerations
Experience with CUDA, Triton, CUTLASS, and distributed training frameworks
Debugging and performance analysis using specialized tools
Curiosity and passion for solving complex problems

Skills mentioned

1

Machine Learning