- New York, New York, United States
Compiler engineer focused on machine learning compilers at Jane Street
Responsibilities
- Shape development of the ML compiler stack
- Design and implement optimizations such as algebraic rewriting, materialize vs recompute tradeoffs, fusion, equality saturation, scheduling languages, loop-level optimization, autotuning
- Develop ML compiler frontends for graph capture, symbolic and dynamic shapes, control flow, automatic differentiation
- Program GPU kernels and architecture using Triton, cuTile, CuTeDSL, CUDA, and tensor cores
- Apply compiler concepts like static analysis, intermediate representations (SSA, CFGs), instruction selection