Impress employers and recruiters.
Choose from hundreds of resume examples.
Impress employers and recruiters.
Choose from hundreds of resume examples.
Tailor your resume to this Machine Learning Compiler Engineer role.
Rezi rewrites your resume against Jane Street's job description. Free.
Tailor your resume to this Machine Learning Compiler Engineer role.
Rezi rewrites your resume against Jane Street's job description. Free.
Don't guess if your resume is good enough.
See how it scores against the Machine Learning Compiler Engineer posting at Jane Street — free, in seconds.
Don't guess if your resume is good enough.
See how it scores against the Machine Learning Compiler Engineer posting at Jane Street — free, in seconds.
About the Role
We are seeking compiler engineers to join our new ML Compilers Team at Jane Street. This role involves shaping the development of our compiler stack to quickly turn research ideas and novel architectures into high-throughput training jobs and low-latency inference. You will help us understand and improve current ML compiler technologies and explore building our own solutions to optimize hardware resources and enable rapid model development and deployment.
Responsibilities
- Shape the development of our compiler stack.
- Understand and improve current ML compiler technologies.
- Explore building our own solutions for ML compilers.
Requirements
- Experience working on an ML compiler is ideal.
- Experience with non-ML compilers is valuable.
- Understanding of how ML frameworks, graph compilers, and kernel DSLs fit together.
- Fluency in English is required.
Skills
- ML frameworks (PyTorch, Jax, TensorFlow)
- Graph compilers (Inductor, XLA, TVM)
- Kernel DSLs (Triton, Helion, cuTile)
- 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)
About the Company
- Jane Street's ever-evolving trading environment serves as a unique, rapid-feedback platform for ML experimentation.
- The ML team consists of individuals with a shared passion for software engineering and designing user-friendly APIs and systems.
- The company welcomes individuals who may not have prior finance experience but possess a curious mind and a passion for solving interesting problems.