Wafer AI - Member of Technical Staff
decircleMillbrae (CA)
Wafer AI - Member of Technical Staff
Posted yesterday
decircleMillbrae (CA)
Software DevelopersComputing Infrastructure Providers, Data Processing, Web Hosting, and Related Services
SENIORITY
Lead
About the role
Wafer AI: maximize intelligence per watt; autonomous AI performance engineers optimizing inference on heterogeneous hardware. 7 people, $4M raised, SF.Comp: $200K-$300K + 0.5%-1% equity.
Location: on-site 5 days FiDi SF.
Visa: Visa transfers (H-1B transfer, OPT); must already be in US; no new H1B.
Responsibilities:
Ship day-zero support for new open-source models tuned for latency/throughputOptimize serving stack: batching, KV cache, speculative decoding, quantizationWrite/tune kernels in CUDA, HIP, Triton for NVIDIA, AMD, TPU, Trainium, D-MatrixDesign, deploy, operate heterogeneous multi-vendor clustersRun production inference: reliability, observability, cost per tokenTech stack: CUDA, HIP, Triton, Python, TPU, AWS TrainiumSource: https://www.paraform.com/share/wafer-ai/cmrtyva4m000j0cl7i7j9wbue1-6 yrs SWE focused on systems close to hardwareGPU kernels (CUDA/HIP/Triton), inference serving optimizationFirst-principles, resourceful, high EQ; customer-facing comfortOn-site 5 days SF; must be in US
Before you apply
Applying takes about a minute. These four things decide how fast it moves after that.
Your profile is current
It's what we read first. Occupations, seniority and locations matter more than a long history.
Two examples you can talk through
Not a portfolio — just two pieces of work where you can explain the decisions and what you'd change.
A number in mind
What you're on now and what would make you move. We negotiate better when we know both.
Your notice period
Employers plan around it, and it's the question that stalls offers most often.
Once you apply, someone reads it and calls you before anything reaches the employer — usually within two working days.