Machine Learning Performance Engineer
Jane Streetabout 5 hours ago
London, United KingdomMid Level / Senior
H1B Sponsor
Responsibilities
- Optimize the performance of machine learning models for training and inference.
- Enhance large-scale training efficiency and ensure low-latency inference.
- Take a whole-systems approach to performance optimization, including storage and networking.
- Debug and analyze training run performance from end to end.
- Evaluate throughput and goodput at the lowest system levels.
Requirements
- Understanding of modern machine learning techniques and toolsets.
- Experience with low-level GPU programming and optimization.
- Knowledge of debugging tools like CUDA GDB and NSight.
- Familiarity with libraries such as Triton, cuDNN, and cuBLAS.
- Background in networking technologies like Infiniband and NVLink.
- Fluency in English.
Tech Stack
Sass
Categories
AI & MLData Engineering