PnR Engineer: AI Inference Mapping Expert

SambaNova, Inc.; Austin (TX); On-site; USD 150,000 - 230,000; Full time

Benefits offered by this job

Health insurance

Dental and Vision

Flexible Spending Account

Well-being programs

Equity

Job summary

SambaNova Systems in Austin, TX, seeks engineers for the Place and Route team to map dataflow graphs onto our Reconfigurable Dataflow Units (RDU). You will develop partitioning, placement, and routing algorithms, collaborating with the compiler, hardware, and performance teams to improve throughput and scalability.

You will model new hardware constraints, tune results for LLM inference at various batch sizes, and ship changes in production releases.

Qualifications

• M.S. or Ph.D. in Computer Science, Computer Engineering, or related field.
• 3+ years of experience developing optimization or graph algorithms in industry or research.
• Expert knowledge of graph algorithms, partitioning algorithms, combinatorial optimization, or MILP solvers.
• Strong proficiency in C/C++
• Familiarity with Python
• Experience designing large, highly parallel, performance-sensitive systems
• Ability to learn hardware architecture and apply it to improve performance
• Experience solving NP-hard problems in practice with heuristics, approximation, or pruned exact methods

Responsibilities

• Develop partitioning, placement, and routing algorithms that meet hardware resource constraints and keep compile times practical
• Tune PnR results for target workloads, including LLM inference at different batch sizes and concurrency levels
• Model new hardware constraints and architecture features in our solvers and heuristics as the chip evolves
• Analyze compiled output with performance engineers and trace throughput gaps back to mapping decisions
• Build the infrastructure for prototyping, regression testing, and deployment of PnR changes
• Work with the compiler and hardware teams to define interfaces and resource models

Skills

C/C++

Python

Optimization algorithms

Graph algorithms

Parallel systems

NP-hard problems

Hardware architecture learning

Education

M.S. or Ph.D. in Computer Science or Computer Engineering

Tools

MLIR

LLVM

TVM

XLA

MILP solvers

Place & Route for FPGAs/CGRA

Job description

SambaNova Systems in Austin, TX, seeks engineers for the Place and Route team to map dataflow graphs onto our Reconfigurable Dataflow Units (RDU). You will develop partitioning, placement, and routing algorithms, collaborating with the compiler, hardware, and performance teams to improve throughput and scalability.

You will model new hardware constraints, tune results for LLM inference at various batch sizes, and ship changes in production releases.