Architecture Modeling Engineer

The role

You will own performance modeling and workload analysis for Arago's optical AI accelerator. You will work across hardware and software teams to evaluate design choices, shape the programming model, and optimize key compute kernels.

What you will do

• Develop fast, flexible performance models to evaluate AI accelerator architecture choices and guide early design decisions.
• Work closely with the software team to characterize key AI workloads, kernels, dataflows, and system bottlenecks.
• Work closely with hardware architects to model compute, memory, interconnect, and scheduling behavior across proposed architectures.
• Drive rapid performance studies and trade-off analyses across throughput, latency, utilization, bandwidth, and efficiency.
• Help define and validate the programming model and workload mapping strategy for the accelerator.
• Work hands-on with the software team to optimize compute kernels and key workloads for the target hardware.
• Build lightweight simulation, profiling, and analysis tools that support fast iteration from architecture exploration through software optimization.

Required qualifications

• Strong mathematical background, particularly in linear algebra, probability, numerical methods, and quantitative performance analysis.
• Strong knowledge of computer architecture, including compute pipelines, memory hierarchies, interconnects, and parallel execution.
• Strong C/C++ and Python programming skills.
• Experience analyzing and optimizing performance-critical kernels or low-level compute workloads.
• Good understanding of AI/ML workloads, tensor operations, data movement, and common execution patterns.
• Ability to work across hardware and software teams and translate workload behavior into architecture and software requirements.
• English at a proficient level.

Nice to have

• Strong industry knowledge about AI hardware, GPUs, memory generations, rack-scale computing.
• Experience building performance models, simulators, or analytical models for CPUs, GPUs, NPUs, or other accelerators.
• Experience with profiling, benchmarking, and identifying hardware or software performance bottlenecks.
• Experience in novel computing architectures: analog computing, in-memory computing, heterogeneous systems.

Benefits

• Competitive cash compensation
• Meaningful stock option plan offered
• Healthcare coverage (including family-friendly options)
• pension contributions
• professional development support
• 25 days of PTO, in addition to public holidays
• Ownership of a key technical domain

About the company

Arago's mission is to re-engineer the foundations of computing from first principles. The company uses proprietary technology that fuses optical and CMOS technologies to deliver an order-of-magnitude increase in performance.