At a Glance
- Tasks: Join our ML Compilers Team to develop and optimise cutting-edge machine learning compilers.
- Company: Jane Street, a leading firm in finance and technology with a collaborative culture.
- Benefits: Competitive salary, flexible working hours, and opportunities for professional growth.
- Other info: Great career growth potential in a supportive and innovative team.
- Why this job: Shape the future of ML technology while working in a dynamic trading environment.
- Qualifications: Experience with ML compilers or related technologies; passion for problem-solving is key.
The predicted salary is between 59400 - 72600 £ per year.
About the position
We’re looking for compiler engineers to join a new ML Compilers Team at Jane Street. Machine learning is a critical pillar of our 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. Our ML team is full of people with a shared love for the craft of software engineering, and for designing APIs and systems that are delightful to use. ML compilers let us quickly turn research ideas and novel architectures into high-throughput training jobs and low-latency inference. To get the most out of our hardware resources and enable rapid model development and deployment, we are working to understand and improve the current ML compiler technologies and explore where it makes sense to build our own solutions. In this role, you would shape the development of our compiler stack. 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.
About you
The ideal candidate has worked on an ML compiler in the past, but non-ML compiler experience is also valuable. Experience with any part of the stack is fine, be it high-level ML frameworks (PyTorch, Jax, TensorFlow), graph compilers (Inductor, XLA, TVM) or kernel DSLs (Triton, Helion, cuTile). You should understand how all those pieces fit together even if you are unfamiliar with specific parts. Some other things you might be excited about are:
- Optimisations in ML compilers: algebraic rewriting, materialise vs recompute tradeoffs and fusion, equality saturation, scheduling languages, loop-level optimisation, autotuning
- ML compiler frontends: graph capture, symbolic and dynamic shapes, control flow, automatic differentiation
- GPU kernel programming and architecture: Triton, cuTile, CuTeDSL, CUDA, compute and memory hierarchies, programming with tensor cores
- Compiler concepts: static analysis, intermediate representations like SSA and CFGs, instruction selection
- ML inference: CUDA Graphs, KV caching, batching
- ML training: Tensor/Sequence/Expert/Pipeline Parallelism, FSDP, NCCL, activation checkpointing
- GPU debugging and optimisation: CUDA-GDB, Nsight Systems, Nsight Compute
- Formal methods: proof assistants, automated theorem provers, agentic proof systems
- Programming languages and DSL design: embedded DSLs, staging and metaprogramming, type systems, functional programming
Fluency in English is required.
Machine Learning Compiler Engineer in London employer: Jane Street
At Jane Street, we pride ourselves on being an exceptional employer, fostering a vibrant work culture that prioritises collaboration and employee engagement. Our London office offers a dynamic environment where the Events Management Specialist will play a crucial role in enhancing team connections through thoughtfully curated events, while benefiting from ample opportunities for professional growth and development. With a strong emphasis on support and innovation, we ensure our employees feel valued and excited to contribute to our collective success.
StudySmarter Expert Advice🤫
We think this is how you could land Machine Learning Compiler Engineer in London
✨Join Local Tech Meetups
Get out there and mingle with fellow developers by joining local tech meetups. It’s a fantastic way to meet people who might be working at Jane Street or know someone who does. Plus, you can pick up some trendy tech skills and trends while you're at it!
✨Contribute to Open Source Projects
Show off your coding chops by jumping into open-source projects. Not only does this give you practical experience, but it also gets you noticed in the dev community. You'll create a killer portfolio that speaks volumes about your skills to Jane Street.
✨Tap into Online Developer Communities
Don’t underestimate the power of online developer communities like GitHub, Stack Overflow, and even Reddit. Participate in discussions, share your projects, and build your visibility. We can often find opportunities through these channels that can lead to a full-time gig at companies like Jane Street.
✨Explore Job Boards Specifically for Tech Roles
Keep your eyes peeled on job boards that focus on tech roles. Sites like TechCareers or Stack Overflow Jobs can often have listings for companies like Jane Street that might not show up on broader job sites. Make it a habit to check these regularly, and don’t hesitate to apply directly through our website!
We think you need these skills to ace Machine Learning Compiler Engineer in London
Some tips for your application 🫡
Show off your coding skills:When applying for a software engineering role, it's super important to showcase your coding skills. Make sure your CV includes your tech stack, any relevant programming languages you’re comfortable with, and examples of projects you've worked on. If you have a GitHub profile, link it up! We love to see code in action.
Tailor your portfolio:For a full-time role, we’d expect to see some solid examples of your work in your portfolio. Make sure to include at least two or three projects that highlight your problem-solving skills and your ability to work with different technologies. Focus on the projects that are most relevant to the position at Jane Street.
Craft a killer cover letter:Your cover letter is your chance to stand out—make it personal! Explain why you want to work at Jane Street and how your skills align with the role. Show us your passion for software development. We dig enthusiastic candidates who understand the value of collaboration and continuous learning!
Be clear and concise:When it comes to writing your CV and cover letter, clarity is key. Avoid jargon that could confuse us and stick to simple, direct language. Highlight your achievements with quantifiable results where possible, and keep everything easy to read. A well-organised application goes a long way!
How to prepare for a job interview at Jane Street
✨Brush Up on Your Coding Skills
For a full-time software engineering role, it's crucial that we stay sharp with our coding abilities. Expect technical questions that might involve solving problems on the spot or discussing algorithms. Practise on platforms like LeetCode or HackerRank to get comfortable with the types of questions that often come up.
✨Know Your Tools and Frameworks
Make sure we’re well-acquainted with the tools and technologies listed in the job description. Familiarise ourselves with any specific frameworks or programming languages mentioned. If Jane Street uses React or Node.js, for instance, be ready to discuss how we’ve used them in previous projects or coursework.
✨Showcase Your Projects
Bring along a portfolio that highlights our best work. This could be code samples, GitHub repositories, or any side projects we’ve built. Make sure we can talk through our thought process for each project, especially the challenges we faced and how we solved them—this shows our problem-solving skills in action.
✨Prepare for Behavioural Questions
While technical skills are key, full-time positions also require cultural fit. Be ready to discuss our previous experiences and how we handle teamwork, conflict, and deadlines. Brush up on the STAR method—Situation, Task, Action, Result—to clearly articulate our past experiences when discussing how we've contributed to a team.
