RefolkCandidates
JobsBuild my resume

Member of Technical Staff (AI Inference Engineer)

Perplexity · London

$65k - $170k/yrStaffEngineeringAI Research & SystemsAI Research & SystemsPosted 7 months ago
Apply with a tailored resumeView the original posting

About this role

We are looking for an AI Inference Engineer to join our growing team. We 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. RESPONSIBILITIES: - 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 kernels 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. - You've built and operated production distributed systems under real load - ideally performance-critical ones. - Comfortable working across languages and layers: Rust for the serving runtime, Python for model code, CUDA/CuteDSL for kernels. - You own problems end-to-end. You can read a research paper on Monday, write a kernel on Wednesday, and debug a production incident on Friday. - Self-directed. You do well in fast-moving environments where the path forward isn't laid out for you. NICE-TO-HAVE: - ML compilers and framework internals: PyTorch internals, torch.compile, cus

Excerpt from the posting Perplexity published. Read the full description on their site before applying.

Skills this posting mentions

BudgetingSystems ProgrammingRust

About Perplexity AI

Perplexity AI builds AI native search tools. Perplexity Ask, a new search interface that uses OpenAI GPT 3.5 and Microsoft Bing to directly answer any question you ask.

All 87 openings at Perplexity AI

Apply to this role, tailored

Queue Member of Technical Staff (AI Inference Engineer) at Perplexity and I will read the posting, rewrite your resume against it, draft the cover letter, and score the fit before you send anything.

Applications finish as drafts. Nothing is sent until you read it and press send. New accounts start with 500 free credits.

Tailor my applicationHow it works

More roles at Perplexity

See all

Similar roles elsewhere

See more

Listed from the job board Perplexity publishes. Refolk is not the employer and does not handle their hiring. Applications go to Perplexity directly.