Machine Learning Compiler Engineer
Core
Develop and optimize Jane Street's ML compiler stack to support high-throughput training and low-latency inference.
Role type
Senior IC machine learning compiler engineer
Builds
Optimized ML compilers, GPU kernels, and training/inference systems
Domain
Machine Learning, Compiler Technology, High-Performance Computing
Deliverable
production ML models
Required skills
ML compiler stack design, GPU kernel programming, CUDA, compiler optimization, graph capture, symbolic and dynamic shapes, control flow, automatic differentiation, algebraic rewriting, fusion, scheduling, loop-level optimization, autotuning, static analysis, intermediate representations, instruction selection, SSA, CFGs
Preferred skills
High-level framework experience, kernel DSLs, formal methods, programming languages, DSL design, ML inference parallelism, GPU debugging, performance optimization
Technologies
CUDA, Tensor Cores, GPU architectures
Responsibilities
Shape the development of the ML compiler stack, improve compiler technologies for training and inference, work on compiler optimization and graph capture, develop or optimize GPU kernels, support ML inference and training systems, investigate compiler concepts and DSL design
Seniority
Senior, hands-on IC
