Machine Learning Compiler Engineer
- Location
- New York, NY
- Listed
- 5 days ago
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
- Optimizations in ML compilers: algebraic rewriting, materialize vs recompute tradeoffs and fusion, equality saturation, scheduling languages, loop-level optimization, 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 optimization: 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
From Jane Street's careers page, first listed on Rolev.ai on November 4, 2026. Rolev.ai is not the employer.
How Rolev.ai helps with this application
Rolev.ai is a job search and application tool for US job seekers. For this Machine Learning Compiler Engineer role at Jane Street, it shows how your resume fits, tailors your resume to the posting, and prepares the application for you to review.
See how you fit
Openings from employer career sites are ranked against your whole resume—and searchable inside Claude, ChatGPT, and Cursor via Rolev.ai MCP.
Tailor without inventing
Every change to your resume is checked against your original. New experience goes in only with your yes—zero fabricated claims.
Draft the full kit
Tailored and Full Match resumes, role-specific cover letters, and answers to employer screening questions are prepared in seconds.
1-Click Extension Autofill
The Chrome extension fills the employer's Greenhouse, Workday, Lever, or Ashby form in 1 click. You review it and press Submit.
Similar roles
- Machine Learning Engineering Manager IISpotify · New York, NY
- Machine Learning Engineer, EXR OPS Tech TeamAmazon.com Services LLC · Bellevue, WA
- Machine Learning EngineerMotorolasolutions · Los Angeles, CA
- Machine Learning ScientistRadiancetech · Ruston, LA
- Machine Learning Engineer LeadRelx · Raleigh, NC
- Machine Learning Engineer (Senior - Senior Staff Level)Wonder · New York, NY
Common questions
Where is the Machine Learning Compiler Engineer job at Jane Street located?
This Machine Learning Compiler Engineer position at Jane Street is located in New York, NY. Always check Jane Street's application page for specific state eligibility or relocation policies.
How do I apply for the Machine Learning Compiler Engineer position at Jane Street?
This listing links directly to Jane Street's Greenhouse application page. You can apply directly on Jane Street's site, or use Rolev.ai to check your resume fit, generate a zero-hallucination tailored resume and cover letter for this posting, and autofill the Greenhouse application form in 1 click with the Rolev.ai Chrome extension. You always review the form and press Submit yourself.