Software Engineer - Machine Learning Frameworks Sydney, Australia
Summary
Building ML frameworks infrastructure—optimizers, code generators, runtime libraries, and orchestration across heterogeneous hardware—using C++, Rust, Python, MLIR, and CUDA at a proprietary trading firm.
The Machine Learning Frameworks team exists for two reasons: making the research iteration loop as fast and frictionless as possible, and making production run as fast as the hardware allows. Both matter, and figuring out how to do both at once is most of what makes the work fun. Concretely, we build optimisers, code generators, runtime libraries, and the orchestration layer that runs computation across heterogeneous hardware. The stack spans C++, Rust, Python, MLIR, and CUDA. You don't need to…