Software Engineer - Machine Learning Frameworks
AI
The Machine Learning Frameworks team at IMC builds optimisers, code generators, runtime libraries, and orchestration layers for heterogeneous hardware. We are looking for a software engineer with production experience in systems languages like C++ or Rust to work across the full compute stack. The role is remote and full-time, based in Sydney, Australia.
What you'll do
- Build optimisers, code generators, runtime libraries, and orchestration layers for heterogeneous hardware.
- Manage the whole compute stack, from high-level pipeline orchestration to low-level hardware behaviour.
- Dissect compute graphs and optimize performance on hot paths.
- Collaborate with stakeholders to understand their actual needs.
What you need
Must-have
C++RustPythonMLIRCUDACode GenerationRuntime LibrariesOrchestration
Soft skills
CollaborationProblem SolvingStakeholder ManagementCommunication
Machine LearningC++RustPythonMLIRCUDAFinancial ServicesRemoteFull-timeEngineering