Machine Learning Performance Engineer

Machine Learning Performance Engineer at Janestreet in London, UK, UK

Position Summary

We are seeking a skilled engineer with experience in low-level systems programming and optimization to join our dynamic Machine Learning team. At Jane Street, machine learning is a critical component of our global business operations. Our ever-evolving trading environment provides a unique platform for rapid ML experimentation, minimizing friction as we incorporate new ideas.

Key Responsibilities

• Optimize the performance of our models, including both training and inference processes.
• Focus on efficient large-scale training, low-latency real-time inference, and high-throughput research environments.
• Improve CUDA performance while adopting a whole-systems approach that includes storage systems, networking, and host- and GPU-level considerations.
• Evaluate the efficiency of our platform at all levels to ensure optimal throughput and minimize latency issues.

Requirements

• A strong understanding of modern ML techniques and toolsets.
• Experience in debugging performance end-to-end during training runs.
• Low-level GPU knowledge, including PTX, SASS, warps, cooperative groups, and more.

About the Company

Jane Street is a leading financial services firm that values curiosity and problem-solving. Many of our team members come from non-finance backgrounds but share a passion for tackling interesting challenges. Our unique environment fosters innovation and continuous learning.

Next Steps

If you're excited about this opportunity, we'd love to hear from you! There's no fixed set of skills, just a curious mind and a desire to solve complex problems. To apply, complete your application directly on this page, or you'll be redirected to the employer's application platform to finish submitting there.