Member of Technical Staff (AI Inference Engineer)
Perplexity · San Francisco, CA · AI
About this role
Perplexity is hiring a staff-level Staff Engineer in the software engineering function based in San Francisco, CA. The posting calls out experience with TensorFlow, PyTorch, LLMs, Deep Learning. Compensation is listed at $220,000–$485,000 per year.
- Role
- Staff Engineer
- Function
- software engineering
- Level
- staff
- Track
- Tech leadership
- Employment
- Full-time
- Location
- San Francisco, CA
- Department
- AI
- Posted
- Apr 13, 2026
More roles at Perplexity
Job description
from Perplexity careersWe build and run the inference engine behind every Perplexity query and deploy dozens of model architectures at scale with tight latency and cost budgets. Our stack is Rust, Python, CUDA, and CuTe DSL - and we need another engineer to join us.
What you will work on
Examples of real work the team does:
New models support. Support transformer-based retrieval, text-generation, and multimodal models in our inference infrastructure, from weight loading, request scheduling and KV-cache management to support in API Gateway.
GPU kernels migration to CuTe DSL. Port our in-house CUDA kernels to NVIDIA's CuTe DSL so they run on GB200 today and are portable to Vera Rubin racks tomorrow.
Rust-native serving runtime. Develop our internal Rust-based inference server to solve all Python pains and keep up with rapidly growing traffic.
Performance optimisation. Profile and fix bottlenecks from network ingress through continuous batching and GPU kernel interleaving.
Reliability and observability. Build dashboards, alerts, and automated remediation so we catch regressions before users do. Respond to and learn from production incidents.
Who we're looking for
Deep experience with GPU programming and performance work (CUDA, Triton, CUTLASS, or similar). Any other deep systems programming experience is a plus.
You understand modern LLM architectures and are able to bring them up reliably in a production environment.
This is an excerpt. Read the full job description on Perplexity careers →