At a Glance
- Tasks: Optimise machine learning models for performance in a fast-paced trading environment.
- Company: Join Jane Street, a leader in finance and technology innovation.
- Benefits: Competitive salary, flexible work options, and opportunities for professional growth.
- Other info: Dynamic team culture with a focus on curiosity and innovation.
- Why this job: Make a real impact by solving complex problems in machine learning.
- Qualifications: Experience in low-level systems programming and modern ML techniques.
The predicted salary is between 60000 - 80000 £ per year.
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-evolving trading environment serves as a unique, rapid-feedback platform for ML experimentation, allowing us to incorporate new ideas with relatively little friction.
Your part here is optimising the performance of our models - both training and inference. We care about efficient large-scale training, low-latency inference in real-time systems and high-throughput inference in research. Part of this is improving straightforward CUDA, but the interesting part needs a whole-systems approach, including storage systems, networking and host- and GPU-level considerations. Zooming in, we also want to ensure our platform makes sense even at the lowest level - is all that throughput actually goodput? Does loading that vector from the L2 cache really take that long?
If you've never thought about a career in finance, you're in good company. Many of us were in the same position before working here. If you have a curious mind and a passion for solving interesting problems, we have a feeling you'll fit right in.
What we're looking for:
- An understanding of modern ML techniques and toolsets.
- The experience and systems knowledge required to debug a training run's performance end to end.
- Low-level GPU knowledge of PTX, SASS, warps, cooperative groups, Tensor Cores and the memory hierarchy.
- Debugging and optimisation experience using tools like CUDA GDB, NSight Systems, NSight Compute.
- Library knowledge of Triton, CUTLASS, CUB, Thrust, cuDNN and cuBLAS.
- Intuition about the latency and throughput characteristics of CUDA graph launch, tensor core arithmetic, warp-level synchronization and asynchronous memory loads.
- Background in Infiniband, RoCE, GPUDirect, PXN, rail optimisation and NVLink, and how to use these networking technologies to link up GPU clusters.
- An understanding of the collective algorithms supporting distributed GPU training in NCCL or MPI.
- An inventive approach and the willingness to ask hard questions about whether we're taking the right approaches and using the right tools.
- Fluency in English.
Machine Learning Performance Engineer in London employer: Job Search Place Limited
At Jane Street, we pride ourselves on being an exceptional employer, offering a dynamic work environment where innovation thrives. Our culture encourages curiosity and collaboration, providing ample opportunities for professional growth within the fast-paced world of finance and machine learning. With a focus on cutting-edge technology and a supportive team atmosphere, you'll find that your contributions directly impact our success while enjoying the benefits of a well-rounded work-life balance in a vibrant location.
StudySmarter Expert Advice🤫
We think this is how you could land Machine Learning Performance Engineer in London
✨Tip Number 1
Network like a pro! Attend industry meetups, conferences, or online webinars related to machine learning and systems programming. You never know who you might bump into that could help you land that dream job.
✨Tip Number 2
Show off your skills! Create a portfolio showcasing your projects, especially those involving CUDA and ML optimisation. This gives potential employers a taste of what you can do and sets you apart from the crowd.
✨Tip Number 3
Prepare for technical interviews by brushing up on your low-level systems knowledge. Practice explaining complex concepts in simple terms, as this will demonstrate your understanding and communication skills.
✨Tip Number 4
Don't forget to apply through our website! We love seeing candidates who are genuinely interested in joining our team. Tailor your application to highlight your experience with ML techniques and systems optimisation.
We think you need these skills to ace Machine Learning Performance Engineer in London
Some tips for your application 🫡
Tailor Your CV:Make sure your CV highlights your experience with low-level systems programming and optimisation. We want to see how your skills align with the role, so don’t be shy about showcasing relevant projects or tools you've worked with!
Craft a Compelling Cover Letter:Your cover letter is your chance to shine! Use it to explain why you're passionate about machine learning and how your background makes you a great fit for our team. We love hearing about your curiosity and problem-solving mindset.
Showcase Your Technical Skills:When filling out your application, be specific about your technical skills. Mention your experience with CUDA, debugging tools, and any library knowledge you have. We’re keen to see how you can optimise performance in real-time systems!
Apply Through Our Website:We encourage you to apply directly through our website. It’s the best way for us to receive your application and ensures you’re considered for the role. Plus, it’s super easy to do!
How to prepare for a job interview at Job Search Place Limited
✨Know Your Tech Inside Out
Make sure you brush up on your knowledge of modern ML techniques and toolsets. Be ready to discuss your experience with CUDA, PTX, and SASS in detail. The more familiar you are with the specifics, the better you'll be able to demonstrate your expertise.
✨Showcase Your Problem-Solving Skills
Prepare to share examples of how you've tackled performance issues in the past. Think about specific instances where you optimised training runs or improved inference times. This will show that you not only understand the theory but can apply it practically.
✨Ask Insightful Questions
Don’t hesitate to ask questions during the interview. Inquire about their current challenges with ML performance or how they approach debugging. This shows your curiosity and willingness to engage with the team’s work, which is something we value highly.
✨Familiarise Yourself with Their Environment
Research Jane Street's trading environment and how it integrates with machine learning. Understanding their unique platform will help you tailor your responses and demonstrate that you're genuinely interested in how you can contribute to their goals.
