Liquid AI · 7 hours ago
Member of Technical Staff - GPU Performance Engineer
San Francisco, CA
Full-time
Hybrid
Mid LevelLiquid AI, spun out of MIT CSAIL, builds general-purpose AI systems that ensure efficiency across various deployment targets. The role focuses on designing and shipping custom CUDA kernels, profiling hardware performance, and integrating research into production code to enhance performance in AI workflows.
Artificial Intelligence (AI)Foundational AIGenerative AIInformation TechnologyMachine Learning
Responsibilities
Write high-performance GPU kernels for our novel model architectures
Integrate kernels into PyTorch pipelines (custom ops, extensions, dispatch, benchmarking)
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
Qualification
Required
Authored custom CUDA kernels (not only calling cuDNN/cuBLAS)
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
Strong C/C++ skills
Preferred
CUTLASS experience and tensor core utilization strategies
Triton kernel experience and/or PyTorch custom op integration
Experience building benchmark harnesses and perf regression tests
Benefits
Health: We pay 100% of medical, dental, and vision premiums for employees and dependents
Financial: 401(k) matching up to 4% of base pay
Time Off: Unlimited PTO plus company-wide Refill Days throughout the year
Company
Liquid AI
Build efficient general-purpose AI at every scale.
Founded in 2023
51-200 employees
H1B Sponsorship
Liquid AI has a track record of offering H1B sponsorships. Please note that this does not
guarantee sponsorship for this specific role. Below presents additional info for your
reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (2)
Funding
Current Stage
Growth StageTotal Funding
$293.1MKey Investors
AMD VenturesOSS Capital L.P.
2024-12-13Series A· $250M
2023-12-01Seed· $37.5M
2023-05-05Seed· $5.6M
Leadership Team
Ramin Hasani
Co-founder and CEO
Mathias Lechner
Co-founder and CTO
Recent News
2025-12-06
Digital Commerce 360
2025-11-15
Company data provided by crunchbase