Software Engineer - Machine Learning Frameworks

IMC B.V.; Sydney; On-site; AUD 120,000 - 180,000; Full time

Job summary

IMC B.V. is building the Machine Learning Frameworks team to accelerate research iterations and production runs across diverse hardware. The stack includes C++, Rust, Python, MLIR, and CUDA, and you will contribute across the compute pipeline from high-level orchestration to low-level optimisations.

You will thrive by moving between languages, understanding what code compiles to, and collaborating with traders and other stakeholders to turn needs into robust, fast software.

Qualifications

• Proficiency in C++, Rust, Python and GPU/accelerator compute.
• Ability to move across languages and understand what code compiles to on hardware.
• Comfort with unfamiliar parts of the stack and learning quickly.

Responsibilities

• Work across the compute stack from high-level orchestration to low-level optimisations.

Skills

C++

Rust

Python

MLIR

CUDA

Job description

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 know all of these things but should be excited about working in this space.

The work is broad. Some of it is high-level, looking at the whole compute pipeline, orchestrating the system and creating a cohesive product. Some of it is in the details, getting into hardware behaviour, dissecting compute graphs, squeezing the last bit of performance out of the hot path. Our job is managing the whole stack, top to bottom.

About you

• You can move between languages without it slowing you down. The right tool matters more to you than your favourite one
• You like understanding what a language compiles down to and how it runs on hardware.
• You're comfortable being handed an unfamiliar part of the stack and working it out.
• You like working with stakeholders across the business and you’re good at figuring out what someone actually needs, not just what they asked for

What you don't need

• A finance background
• Every language we touch
• A particular set of years in any particular role
• We hire engineers who like to solve the problems our traders face - we'll teach you the context on the ground