Wafer AI
Wafer AI

Member of Technical Staff

Full-time
On-site
San Francisco, CA
$200K - $300K
About this role

Member of Technical Staff

Our mission at Wafer is to maximize intelligence per watt by using AI to optimize AI infrastructure, achieving orders of magnitude better energy and cost efficiency per token.


We believe cheap intelligence is the most essential piece of technology for a future of abundance. We care about building a future where intelligence is "too cheap to meter."


Wafer commercializes these efforts by serving serverless and dedicated inference for open source LLMs at the best performance per dollar. Our core bet is doing this through autonomous optimization of heterogeneous hardware.


What you'll do

  • Ship day-zero support for new open-source models, tuned for latency and throughput

  • Optimize the serving stack: batching, KV cache, speculative decoding, quantization

  • Write and tune kernels in CUDA, HIP, and Triton for NVIDIA, AMD, TPU, Trainium, D-Matrix, and more.

  • Design, deploy, and operate heterogeneous clusters across vendors

  • Run production inference across a mixed fleet: reliability, observability, and cost per token at scale


How we evaluate

We score every candidate on seven values:

  1. Infinitely Resourceful

  2. Exceptionalism

  3. Unreasonable Standards

  4. Company Over Self

  5. High EQ

  6. Learns Quickly

  7. First Principles Thinker


How we work

On-site in San Francisco, five days a week. Small team with massive surface area and ownership. You operate with complete autonomy of how to solve problems, and work with the team to set the direction of your work. We don't see engineers as code writers, but as problem solvers. You will do everything from talking to customers to writing custom GPU kernels in esoteric hardware.

in software engineering, with focus on systems close to hardware
Salary
$200K - $300K
Equity
0.5 - 1%
Visa sponsorship details
Open to visa transfers (e.g. OPT, H1B transfers)
Additional visa sponsorship details
Can accommodate most visa types (H-1B transfer, OPT, etc.). Candidates must already be in the US — no overseas sponsorship or new H1-Bs.
On-site work policy
5 days in-office in FiDi, San Francisco
Full-time position
Location
San Francisco, CA
Tech stack
CUDA, HIP, Triton, Python, TPU, AWS Trainium
About Wafer AI

Wafer builds AI agents that work as autonomous performance engineers, optimizing GPU kernels for AI inference.


Our product is serverless and dedicated inference for the world’s fastest open source LLMs, achieved by Wafer's autonomous performance engineers.

Team size

14 employees

Founded

2025

LinkedIn

Visit

Total funding

$40M

Company locations

San Francisco
About the team

Learn more about our team here.

Tech stack
CUDA, HIP, Triton, Python, TPU, AWS Trainium