Staff Compiler Engineer
2 days ago
The Role Were searching for Staff Compiler Engineers to architect and build the ML backend (compiler, run-time, and debugger) for our next-generation OPTUs. You will own integration with PyTorch, Tensorflow, JAX, and MXNet down to our low-level kernel drivers. Your mission will be to create seamless support for a broad ecosystem of large AI models, and ensure we are pushing the limits of their performance envelope by closely partnering with hardware and modelling teams to understand device trade-offs. Responsibilities Ownership: Define and deliver the technical vision and roadmap for your team that unlocks key strategic technical and business goals that are essential to the success of Flux. Collaboration: Partner closely with hardware engineers to align compiler, runtime, and debugger requirements with the OTPU design; ensure software and hardware are designed together to deliver maximum performance. Architect & Build: Design and implement our compiler, runtime, and debugger for PyTorch, TensorFlow, JAX, and MXNet on custom hardware. Optimize Performance: Apply advanced techniques (layout, fusion, scheduling, tiling) to eliminate bottlenecks and maximize throughput. Mentor & Define Standards: Lead code reviews, coach peers, and define best practices in ML backend and performance engineering for your team. Guide Technology Direction: Stay ahead of GPU, AI accelerator, and optical computing trends; propose and prototype innovations. Skills & Experience 7+ years of experience in software engineering with a focus on C/C++ programming. Extensive experience in ML framework internals, compilers, low-level programming, and optimisation techniques. Extensive experience optimising Tensorflow, PyTorch or JAX deep learning models. Extensive experience with multiple toolchains like LLVM, OpenXLA/XLA, MLIR, TVM. 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 work environments. Excellent teamwork and communication skills, with the ability to collaborate effectively with cross-functional teams. Bachelor's degree in computer science, electrical engineering, telecoms engineering, mathematics, or a related field. Personal projects are a key differentiating factor and hold more weight than other requirements. Compensation & Benefits Competitive salary and stock options, youre not just part of the journey, you will own a piece of it. Based in our office in central San Francisco To foster collaboration in our high-growth environment, we require all employees to work from our SF office and live within a 45-minute commute. We offer an extra $24,000/year incentive for those living within 20 minutes. Due to U.S. export control regulations, candidates eligibility to work at Flux depends on their most recent citizenship or permanent residency status. We are generally unable to consider applicants whose most recent citizenship or permanent residence is in certain restricted countries (currently including Iran, North Korea, Syria, Cuba, Russia, Belarus, China, Hong Kong, Macau, and Venezuela). Applicants who have subsequently obtained citizenship or permanent residency in another country not subject to these restrictions may still be eligible. We do not accept unsolicited CVs from recruitment agencies, will not be liable for any fees, and prohibit unauthorised use of our company name in recruitment activities. #J-18808-Ljbffr
-
Staff Compiler Engineer
2 days ago
San Francisco, United States PageBolt WordPress Full timeThe Role We’re searching for Staff Compiler Engineers to architect and build the ML backend (compiler, run-time, and debugger) for our next-generation OPTUs. You will own integration with PyTorch, Tensorflow, JAX, and MXNet down to our low-level kernel drivers. Your mission will be to create seamless support for a broad ecosystem of large AI models,...
-
Staff Compiler Engineer
1 week ago
San Francisco, CA, United States Flux Computing Full timeThe Role We're searching for Staff Compiler Engineers to architect and build the ML backend (compiler, run-time, and debugger) for our next-generation OPTUs. You will own integration with PyTorch, Tensorflow, JAX, and MXNet down to our low-level kernel drivers. Your mission will be to create seamless support for a broad ecosystem of large AI models, and...
-
Staff Compiler Engineer
4 days ago
San Francisco, CA, United States Flux Computing Full timeThe Role We're searching for Staff Compiler Engineers to architect and build the ML backend (compiler, run-time, and debugger) for our next-generation OPTUs. You will own integration with PyTorch, Tensorflow, JAX, and MXNet down to our low-level kernel drivers. Your mission will be to create seamless support for a broad ecosystem of large AI models, and...
-
Staff Compiler Engineer
5 days ago
San Francisco, CA, United States Flux Computing Full timeThe Role We're searching for Staff Compiler Engineers to architect and build the ML backend (compiler, run-time, and debugger) for our next-generation OPTUs. You will own integration with PyTorch, Tensorflow, JAX, and MXNet down to our low-level kernel drivers. Your mission will be to create seamless support for a broad ecosystem of large AI models, and...
-
Staff Compiler Engineer
3 weeks ago
San Francisco, United States Flux Full timeThe Role We’re searching for Staff Compiler Engineers to architect and build the ML backend (compiler, run-time, and debugger) for our next-generation OPTUs. You will own integration with PyTorch, Tensorflow, JAX, and MXNet down to our low-level kernel drivers. Your mission will be to create seamless support for a broad ecosystem of large AI models, and...
-
Staff Compiler Engineer
1 week ago
San Francisco, California, United States Flux Full time £208,000 - £254,000The RoleWe're searching for Staff Compiler Engineers to architect and build the ML backend (compiler, run-time, and debugger) for our next-generation OPTUs. You will own integration with PyTorch, Tensorflow, JAX, and MXNet down to our low-level kernel drivers. Your mission will be to create seamless support for a broad ecosystem of large AI models, and...
-
Compiler Engineer
4 weeks ago
San Francisco, United States Acceler8 Talent Full timeThis range is provided by Acceler8 Talent. Your actual pay will be based on your skills and experience — talk with your recruiter to learn more.Base pay range$200,000.00/yr - $400,000.00/yrWant to build the future compute platform for AGI?Want to build hardware and software to train and run the largest ML workloads for AGI?We are currently looking for...
-
Staff Compiler Engineer: Lead ML Backend
3 weeks ago
San Francisco, United States Flux Full timeA technology company in San Francisco is seeking a Staff Compiler Engineer to architect the ML backend for next-generation models. The role requires expertise in C/C++, ML framework internals, and hardware collaboration. Candidates should have over 7 years of experience and a strong educational background. This position offers a competitive starting salary...
-
Senior Compiler Engineer
2 days ago
San Francisco, United States Flux Computing Full timeSenior Compiler Engineer Join to apply for the Senior Compiler Engineer role at Flux Computing The roleWe’re searching for Senior Compiler Engineers to join the team building the ML backend (compiler, run-time, and debugger) for our next-generation OPTUs that connect PyTorch, Tensorflow, JAX, and MXNet down to our low-level kernel drivers. Your mission...
-
Senior Staff – AI Compiler Engineer
12 hours ago
San Jose, California, United States Advanced Micro Devices, Inc Full timeWHAT YOU DO AT AMD CHANGES EVERYTHING At AMD, our mission is to build great products that accelerate next-generation computing experiences—from AI and data centers, to PCs, gaming and embedded systems. Grounded in a culture of innovation and collaboration, we believe real progress comes from bold ideas, human ingenuity and a shared passion to create...
