companyWafer logo

Technical Staff Engineer - GPU Optimization at Wafer | San Francisco

WaferSan Francisco
On-site Full-time

Clicking Apply Now takes you to AutoApply where you can tailor your resume and apply.


Unlock Your Potential

Generate Job-Optimized Resume

One Click And Our AI Optimizes Your Resume to Match The Job Description.

Is Your Resume Optimized For This Role?

Find Out If You're Highlighting The Right Skills And Fix What's Missing

Experience Level

Experience

Qualifications

Your ResponsibilitiesEnhance and develop our framework for GPU kernel optimization, including multi-turn tool utilization, state management, and reward signals. Create integrations with GPU profilers and compiler toolchains. Architect solutions for remote GPU execution across cloud-based GPUs. Develop trace analysis systems that enable agents to identify performance bottlenecks. Deliver features that engineers utilize daily to optimize infrastructure for leading AI frameworks such as PyTorch, vLLM, NVIDIA, and AMD.

About the job

About the Position

At Wafer, we are on a mission to enhance the intelligence per watt by developing AI systems that can self-optimize. Our journey begins with GPU kernels, and we aim to revolutionize every aspect of ML systems and AI infrastructure. We are a compact, dynamic team of four, supported by renowned investors including Fifty Years, Y Combinator, Jeff Dean, and Woj Zaremba, co-founder of OpenAI. We are seeking passionate engineers eager to innovate at the convergence of AI agents and systems programming.

In this role, you will collaborate closely with our founding team to create the systems that power our GPU optimization platform. Your projects will range from the agent framework that refines kernels to the profiling infrastructure that interfaces with NCU and ROCprofiler, as well as the compiler tools that scrutinize PTX and SASS.

About Wafer

Wafer is dedicated to transforming AI infrastructure, leveraging cutting-edge technology to optimize GPU performance. We pride ourselves on our agile, no-nonsense approach, fostering a culture of rapid innovation and effective problem-solving in a collaborative environment.

Similar jobs

Tailoring 0 resumes

We'll move completed jobs to Ready to Apply automatically.