Liquid AI
Member of Technical Staff - GPU Performance Engineer
San Francisco · Staff+
Sponsorship not specifiedDetected 358 days ago
C++PyTorchResearch
About the role
- We're looking for someone who finds joy in memory hierarchies, tensor cores, and profiler output.
Responsibilities
- Profile and optimize training and inference workflows to eliminate bottlenecks
- Build correctness tests and numerics checks
- Build/maintain performance benchmarks and guardrails to prevent regressions
- Collaborate closely with researchers to turn promising ideas into shipped speedups
- You'll design and ship custom CUDA kernels, profile at the hardware level, and integrate research ideas into production code that delivers measurable speedups in real pipelines (training, post-training, and inference).
Requirements
- Bridges theory and practice: You can translate ideas from papers into implementations that are robust, testable, and performant.
- Strong understanding of GPU architecture and performance: memory hierarchy, warps, shared memory/register pressure, bandwidth vs compute limits
- Proficiency with low-level profiling (Nsight Systems/Compute) and performance methodology
Nice to have
- CUTLASS experience and tensor core utilization strategies
- Triton kernel experience and/or PyTorch custom op integration
- Measurable improvement on at least one critical end-to-end pipeline (throughput and/or latency), validated by repeatable benchmarks
- At least one research-driven technique shipped as a production kernel and maintained over time
- Performance regressions are detectable early via benchmarks/guardrails, not discovered late
- Our team is small, fast-moving, and high-ownership.
Skills
- Our models and workflows require performance work that generic frameworks don't solve.
Compensation
- Competitive base salary with equity in a unicorn-stage company
- Health: We pay 100% of medical, dental, and vision premiums for employees and dependents
- Financial: 401(k) matching up to 4% of base pay
Benefits
- Compensation: Competitive base salary with equity in a unicorn-stage company
- Health: We pay 100% of medical, dental, and vision premiums for employees and dependents
- Time Off: Unlimited PTO plus company-wide Refill Days throughout the year
Apply directly at Liquid AI →Create a free account for alerts like thisView Liquid AI immigration profile
This listing is sourced directly from Liquid AI's careers page and normalized into a canonical job model.