Jobs Companies Liquid AI Member of Technical Staff - GPU Performance Engineer

Über diese Member of Technical Staff - GPU Performance Engineer Stelle bei Liquid AI

Liquid AI · San Francisco

About Liquid AI

Spun out of MIT CSAIL, we build general-purpose AI systems that run efficiently across deployment targets, from data center accelerators to on-device hardware, ensuring low latency, minimal memory usage, privacy, and reliability. We partner with enterprises across consumer electronics, automotive, life sciences, and financial services. We are scaling rapidly and need exceptional people to help us get there.

The Opportunity

Our models and workflows require performance work that generic frameworks don’t solve. 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). Our team is small, fast-moving, and high-ownership. We're looking for someone who finds joy in memory hierarchies, tensor cores, and profiler output.

While San Francisco and Boston are preferred, we are open to other locations.

What We're Looking For

We need someone who:

  • Works profiler-first: You use tools like Nsight Systems / Nsight Compute to find bottlenecks, validate hypotheses, and iterate until improvements show up in end-to-end benchmarks.

  • Bridges theory and practice: You can translate ideas from papers into implementations that are robust, testable, and performant.

  • Executes independently: Given an ambiguous bottleneck, you can drive from profiling to kernel/integration changes to benchmarked results to maintained ownership.

  • Cares about the details: Memory hierarchy, occupancy, launch configs, tensor core utilization, bandwidth vs compute limits.

The Work

  • 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

Desired Experience

Must-have:

  • 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

Nice-to-have:

  • CUTLASS experience and tensor core utilization strategies

  • Triton kernel experience and/or PyTorch custom op integration

  • Experience building benchmark harnesses and perf regression tests

What Success Looks Like (Year One)

  • 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

What We Offer

  • Unique challenges: Our architectural innovations and efficiency requirements offer unique optimization challenges. High ownership from day one.

  • 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

  • Time Off: Unlimited PTO plus company-wide Refill Days throughout the year

Bereit, sich bei Liquid AI zu bewerben?
Bei Liquid AI bewerben

Ähnliche Jobs

Registrieren für Vorschläge, die auf die von Ihnen geöffneten Jobs und gespeicherten Suchen zugeschnitten sind.

Mehr Jobs bei Liquid AI

Alle Jobs bei Liquid AI ansehen →

Jetzt bewerben
🤖

Moment — langsam

JobsRadar wurde für echte Menschen gebaut, die eine schwere Zeit bei der Jobsuche haben — nicht für automatisierte Anfragen. Sie klicken viel zu schnell und sind jetzt vorübergehend blockiert.

Kommen Sie später wieder. Wenn Sie wirklich auf Jobsuche sind, stehen wir hinter Ihnen — verhalten Sie sich einfach wie ein Mensch.

Catch your next role the second it’s posted.

Create a free account and we’ll watch the boards for you — the instant a job matches your search, it lands in your inbox or Telegram. No digging, no refreshing.

Create free account

Free forever · takes 30 seconds · already have one?

Verschaffe dir einen Vorsprung bei der Jobsuche.

Tritt unserem Telegram-Kanal bei für das, was dir hilft, die Stelle zu bekommen — Gehaltsbenchmarks, den wöchentlichen Marktpuls und neue Feature-Drops. Kein Spam, nur Signal.

Dem Kanal beitreten — kostenlos