Software Engineer - Machine Learning Frameworks
Core
Building optimisers, code generators, runtime libraries, and orchestration layers to accelerate research iteration and production compute across heterogeneous hardware.
Role type
Senior IC machine learning framework engineer
Builds
Optimisers, code generators, runtime libraries, and orchestration layers for compute pipelines
Domain
Financial technology / High-performance computing
Deliverable
production ML models
Required skills
C++, Rust, Python, MLIR, CUDA, system-level optimization, compute graph analysis, hardware behavior understanding, cross-language proficiency
Preferred skills
Stakeholder management, translating business needs into technical solutions
Technologies
C++, Rust, Python, MLIR, CUDA
Responsibilities
Designing and implementing optimisers and code generators for ML workloads, developing runtime libraries for heterogeneous hardware, building orchestration layers to manage compute pipelines, optimizing hot paths for maximum performance, collaborating with stakeholders to define technical requirements
Seniority
Senior, hands-on IC
