Member of Technical Staff (AI Inference Engineer)
Perplexity · AI Research & Systems
- Location
- San Francisco, California, United States · Palo Alto, California, United States · New York City, New York, United States
- Metro
- San Francisco Bay Area, New York City
- Workplace
- Unspecified
- Level
- Unspecified
- Type
- Full-time
- Pay
- $220K–$485K/yr (published by employer)
- Posted
- 6 months ago
- Source
- Ashby
Read the full posting & apply ↗
PythonRustKubernetesPyTorchTensorFlowJAXCUDALLMs
What the posting covers
About the company
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 - 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.…
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.…
Good if you touched any of
ML compilers and framework internals: PyTorch internals, torch.compile, custom operators. Distributed GPU communication: NCCL, NVLink, InfiniBand, RDMA libraries, model/tensor parallelism. Low-precision inference: INT8/FP8/FP4 quantization, mixed-precision…
Qualifications
3+ years of professional software engineering experience with meaningful work on ML inference or high-performance systems. Familiarity with at least one deep learning framework (PyTorch, JAX, TensorFlow). Understanding of GPU architectures (memory hierarchy,…
Excerpts only. The employer’s page has the complete posting.