Compiler Engineer - Contractor
📌London, United Kingdom 🇬🇧
⏱︎ contract
🧙♂️ senior
The Contract
We’re seeking a highly experienced and motivated Compiler Engineer [contractor] for a 6 month engagement outside of IR-35 to work on our ML Backend and functional simulator for our next-generation OTPUs with our permanent team. Your mandate will be to deliver a production-ready ML backend (compiler, runtime, debugger) and functional simulator that can accurately run most large AI models.
This role demands deep expertise in C and C++ programming, ML framework internals, compiler construction, and optimisation techniques.
Responsibilities
- Implement Runtime: Build a runtime that seamlessly integrates with PyTorch, JAX, and TensorFlow (PJRT) for both training and inference execution patterns. The runtime must support asynchronous execution and multiple devices.
- Implement Compiler: Build a compiler that is extensible to future optimisation patterns across operation fusion, layout optimisation, tiling, and scheduling.
- Implement Debugger & Diagnostics: Support optional runtime assertions and compile-time dumps, tensorboard timelines, and JAX I/O callbacks.
-
Implement Functional Simulator: Build a functional simulator that mocks our kernel-space driver, allowing the software team to lower operations ahead of hardware teams.
- 5+ years of experience in software engineering with a focus on C/C++ programming.
- Extensive experience in compilers, low-level programming, and optimisation techniques.
- Practical experience applying machine learning in high-performance computing contexts.
- Strong problem-solving skills and the ability to think critically and creatively.
- Experience in high-pace, dynamic environments.
- Bachelor's degree in computer science, electrical engineering, telecoms engineering, mathematics, or a related field.
-
Excellent communication skills, with the ability to collaborate effectively with cross-functional teams.
We’re building fast and that includes our benefits. More exciting additions are coming soon for the Flux crew.
If you are passionate about pushing the boundaries of what's possible in AI and thrive in a high-energy, fast-paced environment, we want to hear from you. Apply now to join Flux and be a key player in shaping the future of computing.
Other similar jobs
Compiler Developer
@ IBM, 📍United Kingdom 🇬🇧
Senior Compiler Engineer - Backend
@ NVIDIA, 📍United Kingdom 🇬🇧
Senior Compiler Engineer - Backend
@ NVIDIA, 📍United Kingdom 🇬🇧
Senior Compiler Engineer - Backend
@ NVIDIA, 📍United Kingdom 🇬🇧
Senior Compiler Engineer - Backend
@ NVIDIA, 📍United Kingdom 🇬🇧
Staff GPU Compiler Engineer
@ Qualcomm, 📍United Kingdom 🇬🇧
Software Dev Engineer - Compiler, Annapurna Labs
@ Amazon, 📍United Kingdom 🇬🇧
Staff GPU Compiler Engineer
@ Qualcomm, 📍United Kingdom 🇬🇧
Sr ML Compiler Engineer , Annapurna Labs
@ Amazon Web Services (AWS), 📍United Kingdom 🇬🇧
GPU compiler Engineer
@ Qualcomm, 📍United Kingdom 🇬🇧
-
Employment
⏱︎ contract
-
Experience
🧙♂️ senior
-
Skills
-
Industry
-
Find similar jobs
Compiler Developer
@ IBM, 📍United Kingdom 🇬🇧
Senior Compiler Engineer - Backend
@ NVIDIA, 📍United Kingdom 🇬🇧
Senior Compiler Engineer - Backend
@ NVIDIA, 📍United Kingdom 🇬🇧
Senior Compiler Engineer - Backend
@ NVIDIA, 📍United Kingdom 🇬🇧
Senior Compiler Engineer - Backend
@ NVIDIA, 📍United Kingdom 🇬🇧
Staff GPU Compiler Engineer
@ Qualcomm, 📍United Kingdom 🇬🇧
Software Dev Engineer - Compiler, Annapurna Labs
@ Amazon, 📍United Kingdom 🇬🇧
Staff GPU Compiler Engineer
@ Qualcomm, 📍United Kingdom 🇬🇧
Sr ML Compiler Engineer , Annapurna Labs
@ Amazon Web Services (AWS), 📍United Kingdom 🇬🇧
GPU compiler Engineer
@ Qualcomm, 📍United Kingdom 🇬🇧