About this role

Join IMC's Machine Learning Frameworks team to accelerate research and optimize production performance. You will build optimizers, code generators, and runtime libraries for heterogeneous hardware, working across C++, Rust, and Python. This role involves managing the full compute stack, from high-level orchestration to low-level hardware optimization.

What we are looking for

6

Build ML optimizers, code generators, and runtime libraries
Optimize computation across heterogeneous hardware
Work with C++, Rust, Python, MLIR, and CUDA
Manage the entire compute stack from top to bottom
Collaborate with stakeholders to understand and deliver needs
Focus on performance and research iteration speed

Skills mentioned

4

PythonC++RustMachine Learning