Flux Computing ยท 1 month ago
Staff Compiler Engineer
San Francisco, CA
Full-time
Onsite
Senior Level, Lead/Staff
$275K/yr - $336K/yr
7+ years expFlux Computing is seeking Staff Compiler Engineers to architect and build the ML backend for their next-generation OPTUs. The role involves defining technical visions, collaborating with hardware engineers, and optimizing performance for a wide range of AI models.
Artificial Intelligence (AI)HardwareMachine LearningManufacturingOptical Communication
No H1B
U.S. Citizen Only
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
Qualification
Required
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
Preferred
Personal projects are a key differentiating factor and hold more weight than other requirements
Benefits
Generous stock options in a rapidly growing AI company
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.
Company
Flux Computing
Flux Computing designs optical AI accelerators that use light-based processors for training and inference on large models.
51-200 employees
Funding
Current Stage
Growth StageCompany data provided by crunchbase