Jobs in the United States

25+ jobs

Save this search

Added 13 days ago
Machine Learning - Compiler Engineer , AWS Neuron, Annapurna Labs
Amazon
Cupertino, CA
This role is for a software engineer in the Compiler team for AWS Neuron. As part of this role, you will be responsible for building next generation Neuron compiler which transforms ML…
Added 13 days ago
Software Development Manager - Compiler
Amazon
Cupertino, CA
…innovation. The Inferentia chip delivers best-in-class ML inference performance at the lowest cost in cloud. Trainium delivers the best-in-class ML training performance with the most…
Added 13 days ago
Senior AI SoC Modeling Engineer, Annapurna Labs Machine Learning Accelerators, AWS
Amazon
Cupertino, CA
…builds C++ models of these custom SoCs that RTL designers, verification engineers, and software teams depend on throughout the silicon development lifecycle. We're looking for a modeling…
Added 13 days ago
AI SoC Modeling Engineer, Annapurna Labs Machine Learning Accelerators, AWS
Amazon
Cupertino, CA
…builds C++ models of these custom SoCs that RTL designers, verification engineers, and software teams depend on throughout the silicon development lifecycle. We're looking for a modeling…
Added 13 days ago
Software Development Manager - NKI Compiler, AWS Neuron, Annapurna Labs
Amazon
Cupertino, CA
…development team. As an SDM III, you will lead a team of experienced compiler engineers developing compiler optimization algorithms and deploying, at scale, a new compiler targeting AWS…
Added 13 days ago
Machine Learning - Compiler Engineer , AWS Neuron, Annapurna Labs
Amazon
Cupertino, CA
This role is for a software engineer in the Compiler team for AWS Neuron. As part of this role, you will be responsible for building next generation Neuron compiler which transforms ML…
Added 13 days ago
Machine Learning - Compiler Engineer , AWS Neuron, Annapurna Labs
Amazon
Seattle, WA
This role is for a software engineer in the Compiler team for AWS Neuron. As part of this role, you will be responsible for building next generation Neuron compiler which transforms ML…
Added 16 days ago
SRAM Design Engineer
NVIDIA USA
CA
SRAM compiler development: Envisioning, defining, and coding more efficient ways to automate the simultaneous assembly and validation of multiple unique SRAM macros using NVIDIA's extensive…
Added 18 days ago
ML Systems Research Engineer, RL / Inference / Agent Systems
Advanced Micro Devices, Inc
Santa Clara, CA
You will work across compute optimization, hardware engineering automation, verification, simulation, debugging. The emphasis is on scalable ML systems that make research practical,…
Added 19 days ago
Sr. AI/ML Platform Engineer
Advanced Micro Devices, Inc
Santa Clara, CA
Partner with Applied AI Engineers to make Blueprint harnesses reusable across kernel optimization, RTL optimization, verification, firmware, and CPU/GPU performance workflows.
Added 19 days ago
AI Engineer, Recursive Self-Improvement for Compute
Advanced Micro Devices, Inc
Santa Clara, CA
We are hiring AI Engineers to build recursive self-improvement systems for compute. This role sits at the intersection of AI systems, performance engineering, hardware-aware optimization,…
Added 19 days ago
Applied AI Engineer, Top Hardware Priorities
Advanced Micro Devices, Inc
Santa Clara, CA
You will work across hardware-adjacent domains such as design optimization, verification, simulation, firmware, performance debugging, routing, issue triage, and compute library…
Added 19 days ago
Software Development Engineer - Rust, Compilers, and GPU Systems (2)
Advanced Micro Devices, Inc
San Jose, CA
Rust is not incidental to this role. You will help establish how Rust is used in performance-sensitive and high-trust parts of the GPU stack, including the compiler and tooling support,…
Added 19 days ago
Staff Software Development Engineer - Rust, Compilers, and GPU Systems (1)
Advanced Micro Devices, Inc
San Jose, CA
Rust is not incidental to this role. You will help establish how Rust is used in performance-sensitive and high-trust parts of the GPU stack, including the compiler and tooling support,…
Added 19 days ago
Staff Software Development Engineer - Rust, Compilers, and GPU Systems (3)
Advanced Micro Devices, Inc
San Jose, CA
Rust is not incidental to this role. You will help establish how Rust is used in performance-sensitive and high-trust parts of the GPU stack, including the compiler and tooling support,…
Added 19 days ago
SF
Member of Mathematical Staff
San Francisco Tensor Company
San Francisco, CA
- $225,000 to $315,000 USD / year
AT SF TENSOR, WE'RE BUILDING THE FUTURE OF HIGH-PERFORMANCE COMPUTE
Added 19 days ago
M
Senior Silicon Emulation Engineer
Mythic
Austin, TX
- $120,000 to $225,000 USD / year
Mythic's Digital Design Team is seeking a Silicon Emulation Engineer to support the development and validation of AI accelerator and AI-centric SoCs using hardware emulation platforms. This…
Added 19 days ago
Model Bring-up Engineer / ML Compiler Engineer
General Compute
San Francisco, CA · NYC, NY
General Compute is the neocloud for alternative chips.
Added 19 days ago
Machine Learning - Compiler Engineer , AWS Neuron, Annapurna Labs
Annapurna Labs (U.S.) Inc.
Cupertino, CA
This role is for a software engineer in the Compiler team for AWS Neuron. As part of this role, you will be responsible for building next generation Neuron compiler which transforms ML…
Added 19 days ago
Software Development Manager - Compiler
Annapurna Labs (U.S.) Inc.
Cupertino, CA
…innovation. The Inferentia chip delivers best-in-class ML inference performance at the lowest cost in cloud. Trainium delivers the best-in-class ML training performance with the most…
Added 19 days ago
Senior AI SoC Modeling Engineer, Annapurna Labs Machine Learning Accelerators, AWS
Annapurna Labs (U.S.) Inc.
Austin, TX · Cupertino, CA
…builds C++ models of these custom SoCs that RTL designers, verification engineers, and software teams depend on throughout the silicon development lifecycle. We're looking for a modeling…
Added 19 days ago
AI SoC Modeling Engineer, Annapurna Labs Machine Learning Accelerators, AWS
Annapurna Labs (U.S.) Inc.
Austin, TX · Cupertino, CA
…builds C++ models of these custom SoCs that RTL designers, verification engineers, and software teams depend on throughout the silicon development lifecycle. We're looking for a modeling…
Added 19 days ago
Software Development Manager - NKI Compiler, AWS Neuron, Annapurna Labs
Annapurna Labs (U.S.) Inc.
Cupertino, CA
…development team. As an SDM III, you will lead a team of experienced compiler engineers developing compiler optimization algorithms and deploying, at scale, a new compiler targeting AWS…
Added 19 days ago
Machine Learning - Compiler Engineer , AWS Neuron, Annapurna Labs
Annapurna Labs (U.S.) Inc.
Seattle, WA · Cupertino, CA
This role is for a software engineer in the Compiler team for AWS Neuron. As part of this role, you will be responsible for building next generation Neuron compiler which transforms ML…
Added 19 days ago
Machine Learning - Compiler Engineer , AWS Neuron, Annapurna Labs
Annapurna Labs (U.S.) Inc.
Cupertino, CA
This role is for a software engineer in the Compiler team for AWS Neuron. As part of this role, you will be responsible for building next generation Neuron compiler which transforms ML…

General Compute

Model Bring-up Engineer / ML Compiler Engineer

San Francisco, CA, United States
NYC, NY, United States

Apply About employer

Applications open in a new tab.

Job details

Posting language: English
Posted: Sep 1, 2026

Original job description

ABOUT US

General Compute is the neocloud for alternative chips.

Inference is fragmenting: purpose-built silicon from SambaNova, Cerebras, Positron, d-Matrix, and others already beats GPUs on decode, and we productionize that hardware - we buy the racks, find the data center space, and run it for our customers. Each piece of hardware runs the workload it's actually built for: prefill stays on GPUs, decode moves to the chip built for it, and today that means generating tokens 5–7× faster than existing GPU-based competitors. Our customers are frontier labs, fast-growing AI application companies, and asset-light clouds.

We closed a $15M seed round in May 2026, and have since closed a $400M debt facility - $100M funded upfront by Upper90, with the balance available for drawdown - collateralized by our inference chips.

ABOUT THE ROLE

You'll take a new model and get it running - correctly - on our ASIC in record time. When a frontier model drops, the only question that matters is how fast we can land it on our silicon and start serving it. You own that loop: from reference weights, through the compiler, to first correct tokens. The low-level runtime is co-owned with our hardware partner today; your job is everything it takes to get a brand-new architecture compiled, verified, and fast on top of it.

The bet of this role is that bring-up should be an agentic loop, not a hand-port. You'll build the harness of agents that compiles, runs, diffs against reference, and localizes failures - so the marginal model comes up faster than the last one did. Correctness first, optimization second: get it right, prove it's right, then make it cheap. This is a senior IC role on a small team. You'll own the bring-up pipeline, not tickets.

What You'll Do:

Own model bringup end-to-end. Take a new architecture - a frontier LLM, an MoE, a multimodal model - from reference weights to first correct tokens running on our ASIC, in days, not quarters.
Build the agentic bringup loop. The differentiator isn't hand-porting one model - it's the harness of agents that compiles, runs, diffs against reference, localizes the failing op, and iterates without you in the inner loop. Each model you land should make the loop better at landing the next one.
Live in the compiler. Graph capture, IR lowering, op coverage, kernel selection - when a model won't compile or produces wrong numbers, the fix is yours, whether it's a missing lowering, a fused-kernel bug, or a numerics mismatch.
Own correctness before speed. Build the verification harness - layer-by-layer activation diffs, logit parity, end-to-end evals - that proves a freshly brought-up model matches reference before anyone trusts a token of it.
Then optimize. Once it's correct, make it fast: operator fusion, quantization, memory layout, batching and KV-cache behavior on our hardware. Bringup gets it running; this is where it earns its cost-per-token.
Work shoulder-to-shoulder with our hardware partner's compiler and runtime team. You're the person who turns 'the chip can technically run this' into 'this model is live and correct in production.

What we need from you:

5+ years in systems or ML systems, with real depth in at least one of: ML compilers, model porting/bringup, or high-performance kernels.
You've taken a model architecture you didn't design and made it run - and run correctly - on a target it wasn't written for. Numerics debugging doesn't scare you.
Strong on the internals of modern LLM inference: transformers, attention, KV cache, MoE routing, quantization, batching. You can read a new model's reference implementation and know what will be hard to lower.
Comfortable inside a compiler stack - MLIR/LLVM, XLA, or a vendor graph compiler - at the level of IR, lowering, and op coverage, not just calling into one.
Fluent with agentic tooling. You'd rather build the agent that runs the tedious bringup loop than run it by hand - and you have the taste to know where the loop still needs a human.
Self-directed. We don't assign tickets - you'll see the next model coming and have it half brought-up before anyone asks.

Nice-to-haves:

Have worked on a non-NVIDIA accelerator - TPU, Trainium/Inferentia, Tenstorrent, Groq, Cerebras, or similar - at the compiler or model-bringup layer.
Kernel-level experience in CUDA, Triton, or a vendor kernel language. You know why a fused attention kernel beats three unfused ops.
Have built eval and numerics-verification harnesses (logit parity, activation diffing) for models in production.
Contributed to a graph compiler or serving runtime - XLA, TVM, MLIR, vLLM, TGI, TensorRT-LLM, or SGLang.
Have built agent loops or LLM-driven tooling that did real engineering work, not demos.