15 часов назад
Staff Modeling Architect (AI)
250 000 - 290 000$
Мэтч & Сопровод
Для мэтча с этой вакансией нужен Plus
Описание вакансии
Текст:
TL;DR
Staff Modeling Architect (AI): Building functional, performance, power, and energy models for photonic AI accelerators with an accent on hardware/software co-design, workload modeling, and simulation fidelity. Focus on connecting production AI workloads to programming models, optimizing tiling and memory hierarchies, implementing bit-accurate C++ models, and correlating simulations with RTL.
Location: Austin, Texas or Sunnyvale, California; full-time onsite position
Salary: $250,000–$290,000 annually, plus equity
Company
develops silicon photonics and programmable metasurface technology for energy-efficient, high-performance AI inference hardware.
What you will do
- Bring up dense, Mixture of Experts, attention, retrieval, speech, vision, recommendation, and other inference workloads on the accelerator model stack.
- Bind Hugging Face and PyTorch workloads to the programming model and runtime so software and architecture share the same functional behavior.
- Co-design tiling, scheduling, ISA, SRAM and HBM hierarchy, NoC traffic, and multi-chip mapping across pipeline, tensor, and sequence parallelism.
- Develop Python energy, latency, performance, power, and area models, including roofline analysis and design-space exploration.
- Implement bit-accurate C++ functional models and contribute to the event-driven simulation kernel, including cycle-accurate and cycle-approximate simulation.
- Define modeling methodology for a workload area, align models with RTL through Verilator and SystemVerilog co-simulation, and mentor modeling engineers.
Requirements
- 8+ years of experience in hardware, functional, performance, or accelerator modeling and simulation.
- Strong grounding in computer architecture, microarchitecture, memory systems, and AI accelerators.
- Modern C++17 or later and Python with NumPy, Pandas, and Matplotlib.
- Experience working inside a discrete-event, cycle-approximate, or cycle-accurate simulator such as SystemC, gem5, SST, or a custom kernel.
- Ability to build LLM or accelerator workloads covering prefill, decode, MoE, GEMM tiling, and quantization.
- BS, MS, or PhD in Computer Engineering, Electrical Engineering, Computer Science, or equivalent practical experience.
Nice to have
- PhD in Computer Engineering, Electrical Engineering, or Computer Science.
- Hardware/software co-design experience with MLIR, TVM, XLA, ONNX, operator fusion, or graph compilers.
- Experience extending simulation kernels or correlating analytical models with silicon, datasheets, or measured accelerators.
- Familiarity with TLM 2.x, Verilator, SystemVerilog, DPI, UVM, HBM, DRAM controllers, cache, SRAM, NoC, AXI, DMA, and scratchpad memory.
- Power modeling with McPAT or CACTI, FPGA prototyping, or hardware emulation.
Culture & Benefits
- Work on photonics and AI hardware at an early-stage company developing energy-efficient inference technology.
- Health plan premiums are fully covered for employees and dependents, with HSA contributions.
- Unlimited PTO and flexible benefits selection.
- 401(k) matching, stock options, dental, vision, life, hospital, critical illness, and accident insurance.
Будьте осторожны: если работодатель просит войти в их систему, используя iCloud/Google, прислать код/пароль, запустить код/ПО, не делайте этого - это мошенники. Обязательно жмите "Пожаловаться" или пишите в поддержку. Подробнее в гайде →
Похожие вакансии
2 дня назад
Lead Vector Compute Architect (GPU)
250 000 - 350 000$
5 дней назад
Senior Engineer, Architecture & Performance Research Engineer for Data Center and Agentic AI CPU
138 000 - 206 000$
2 дня назад
Senior Solutions Architect / Sales Engineer (HPC & Computational Engineering)
175 000 - 200 000$
5 дней назад
Senior Director, Architecture Research Lab (AI)
246 000 - 430 000$
5 дней назад
Principal Engineer, Architecture & Performance Research Engineer for Data Center and Agentic AI CPU (RISC-V)
219 000 - 351 000$
4 дня назад
Agent Architect (AI)
125 000 - 220 000$