Flux Computing ยท 1 month ago
Senior Performance Modelling Engineer
San Francisco
Full-time
Onsite
Senior Level
$230K/yr - $282K/yr
5+ years expFlux Computing is a rapidly growing AI company seeking a Senior Performance Modelling Engineer to create and own analytical and simulation models for OTPU architecture and software evolution. The role involves building functional simulators and cycle-accurate models, collaborating with various teams, and delivering insights that influence software and hardware roadmaps.
Artificial Intelligence (AI)HardwareMachine LearningManufacturingOptical Communication
No H1B
U.S. Citizen Only
Responsibilities
Own and deliver projects on your team's roadmap that unlock key high-impact technical and business milestones that drive the success of Flux
Work shoulder-to-shoulder with hardware, compiler and ML framework teams to ensure models reflect reality and reality meets performance goals
Design, build, and maintain a functional simulator of the OPTU subsystem and full pipeline
Design and maintain architectural & cycle-accurate models of the OPTU subsystems and pipeline. Identify throughput, latency and utilisation hot-spots; propose architectural, or scheduling fixes
Instrument benchmarks (LLMs, diffusion, graph workloads) to collect detailed traces
Run massive parameter sweeps with your functional and to understand tradeoffs and guide the software, hardware, and optical teams. Package results into clear, quantitative analyses and design recommendations
Develop Python/C++ tooling for trace parsing, statistical analysis and visualisation. Integrate models into CI so that every RTL commit gets a performance smoke test
Qualification
Required
5+ years building performance or power models for CPUs, GPUs, ASICs, or accelerators
Strong coding ability in C++ and Python; experience with discrete-event or cycle-accurate simulators (e.g., gem5, SystemC, custom in-house)
Solid grasp of computer-architecture fundamentals: memory systems, interconnects, queuing theory, Amdahl/Gustafson analysis
Familiarity with machine-learning workloads and common frameworks (PyTorch, TensorFlow, JAX)
Comfort reading RTL or schematics and discussing micro-architectural trade-offs with hardware designers
Excellent data-visualisation and communication skills: able to turn millions of simulation samples into one decisive slide
Bachelor's+ in EE, CS, Physics, Applied Maths or related
Preferred
Advanced degree preferred but not required
Personal or open-source projects in simulators, ML kernels, or performance analysis are a significant plus
Benefits
Generous stock options in a rapidly growing AI company
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