Staff Performance Modelling Engineer

Posted 10 Days Ago
Be an Early Applicant
San Francisco, CA
In-Office
275K-335K Annually
Senior level
Artificial Intelligence • Software • Semiconductor
The Role
As a Staff Performance Modelling Engineer, you'll lead the creation of analytical models to enhance OTPU architecture while collaborating with engineering teams to meet performance goals.
Summary Generated by Built In

The Role

We’re searching for a Staff Performance Modelling Engineer (San Francisco), to create and own the analytical and simulation models that steer OTPU architecture and software evolution. You will build functional simulators as well as high-fidelity, cycle-accurate models of our optical compute system. This role is critical to explore “what-if” design spaces, and deliver insights that directly influence our software, hardware, and optical roadmaps. This role sits at the crossroads of hardware architecture, software tooling and machine-learning workload analysis, perfect for an engineer who loves data-driven decision-making and fast iteration.

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 all engineering teams to help shape our overall system architecture and delivery while ensuring models reflect reality and reality meets performance goals.

  • Champion Modelling: Educate peers on modelling methodology and champion data-driven design culture.

  • Functional Simulator: Design, build, and maintain a functional simulator of the OPTU subsystem and full pipeline.

  • Performance Simulator: 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.

  • Workload Analysis & Bottleneck Hunting: Instrument benchmarks (LLMs, diffusion, graph workloads) to collect detailed traces.

  • Design-Space Exploration: Run massive parameter sweeps with your functional and to understand tradeoffs and guide the software, hardware, and optical teams.

  • Tooling & Automation: 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.

Skills & Experience

  • 7+ years building performance or power models for CPUs, GPUs, ASICs, or accelerators.

  • Proven track record providing technical leadership to a team of 5~10 engineers, resulting in significant business impact.

  • Strong coding ability in C++ and Python; experience with discrete-event or cycle-accurate simulators (e.g., gem5, SystemC, custom in-house).

  • Strong 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; advanced degree preferred but not required.

  • Personal or open-source projects in simulators, ML kernels, or performance analysis are a significant plus.

Compensation & Benefits

  • Competitive salary and stock options, you’re 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.

Top Skills

C++
Gem5
Jax
Python
PyTorch
Systemc
TensorFlow
Am I A Good Fit?
beta
Get Personalized Job Insights.
Our AI-powered fit analysis compares your resume with a job listing so you know if your skills & experience align.

The Company
40 Employees
Year Founded: 2024

What We Do

Flux Computing designs and manufactures optical processors to train and run inference on large AI models.

Similar Jobs

Airwallex Logo Airwallex

Associate Director, Regulatory Legal, Americas

Artificial Intelligence • Fintech • Payments • Financial Services • Generative AI
In-Office
2 Locations
1800 Employees
200K-280K Annually

Airwallex Logo Airwallex

Director, Regulatory Legal, Americas

Artificial Intelligence • Fintech • Payments • Financial Services • Generative AI
In-Office
2 Locations
1800 Employees
250K-320K Annually

Gusto Logo Gusto

Senior Product Manager

Fintech • HR Tech
Easy Apply
Remote or Hybrid
6 Locations
4405 Employees
152K-230K Annually

Crunchyroll Logo Crunchyroll

Recruiting Coordinator

Digital Media • eCommerce • Gaming • Mobile • News + Entertainment
Hybrid
Los Angeles, CA, USA
1300 Employees
30-40 Annually

Similar Companies Hiring

PRIMA Thumbnail
Travel • Software • Marketing Tech • Hospitality • eCommerce
US
15 Employees
Scotch Thumbnail
Software • Retail • Payments • Fintech • eCommerce • Artificial Intelligence • Analytics
US
25 Employees
Idler Thumbnail
Artificial Intelligence
San Francisco, California
6 Employees

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account