Jane Street · London, England, United Kingdom

Machine Learning Compiler Engineer

最近核查時仍有刊登最近確認刊登:2026-11-05 18:37:35.57261+00

招聘狀態可能變動,申請前請查看原始職位。

薪酬未公開
學歷未说明
語言英语 · 未指定等级
經驗未说明

來源發布日期:2026-11-03T15:16:08-05:00

職位要求

  • Fluency in English is required
查看招聘來源的要求摘錄

原始資料於 2026-11-04T13:50:42.608293+00:00 查閱;內容可能已更新,請查看原始職位。

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
If you're a recruiting agency and want to partner with us, please reach out to [email protected].

來源:公司官网 · 資料查閱日期 2026-11-04T13:50:42.608293+00:00。職位狀態及要求請以原始招聘頁面為準。