> Markdown version of [/jobs/ext/2256013-ai-inference-engineer-gpu-scale-rust-python-equity](https://www.wearedevelopers.com/jobs/ext/2256013-ai-inference-engineer-gpu-scale-rust-python-equity). Every page supports `.md` or `Accept: text/markdown`. Links point to the HTML versions so they work for humans too. Agent guide: [/agents.md](https://www.wearedevelopers.com/agents.md). --- # AI Inference Engineer | GPU-Scale Rust/Python | Equity - **Company:** Perplexity - **Location:** Greater London, UK - **Experience:** Experienced - **Contract:** Permanent contract - **Skills:** Artificial Intelligence, Profiling, Nvidia CUDA, Software Debugging, Distributed Systems, InfiniBand, Python (Programming Language), Performance Tuning, Remote Direct Memory Access, Tensorflow, Software Engineering, System Programming, Rust (Programming Language), Pytorch, Large Language Models, Deep Learning, Gpu Programming, Syntactically Awesome Style Sheets (SASS), Kubernetes, Api Gateway - **Published:** August 26, 2026 - **Apply:** https://www.collegerecruiter.com/job/2815189535-ai-inference-engineer--gpu-scale-rustpython--equity ## About the Role * 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, custom operators. * Distributed GPU communication: NCCL, NVLink, InfiniBand, RDMA libraries, model/tensor parallelism. * Low-precision inference: INT8/FP8/FP4 quantization, mixed-precision serving. * Profiling and debugging tools: Nsight Compute/Systems, CUDA-GDB, PTX/SASS analysis. * Container orchestration: Kubernetes, GPU scheduling, autoscaling inference workloads., * 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, warp scheduling, tensor cores). * Understanding of common LLM architectures and inference optimization techniques (e.g. quantization, speculative decoding, prefill-decode disaggregation). Final offer amounts are determined by multiple factors including experience and expertise. ## Description 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, custom operators. * Distributed GPU communication: NCCL, NVLink, InfiniBand, RDMA libraries, model/tensor parallelism. * Low-precision inference: INT8/FP8/FP4 quantization, mixed-precision serving. * Profiling and debugging tools: Nsight Compute/Systems, CUDA-GDB, PTX/SASS analysis. * Container orchestration: Kubernetes, GPU scheduling, autoscaling inference workloads. 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, warp scheduling, tensor cores). * Understanding of common LLM architectures and inference optimization techniques (e.g. quantization, speculative decoding, prefill-decode disaggregation). Final offer amounts are determined by multiple factors including experience and expertise. Equity: In addition to the base salary, equity may be part of the total compensation package. ## Related Videos - [The Gashlycrumb Tinies of AI Networking You Must Know (or Languish!)](https://www.wearedevelopers.com/videos/2067-the-gashlycrumb-tinies-of-ai-networking-you-must-know-or-languish) - [Tour de Force: Open-Source LLM Inference Optimization from Simple to Sophisticated](https://www.wearedevelopers.com/videos/100099-tour-de-force-open-source-llm-inference-optimization-from-simple-to-sophisticated) - [Profiling Symfony & PHP apps with Blackfire](https://www.wearedevelopers.com/videos/265-profiling-symfony-php-apps-with-blackfire) - [Photonic Computing: Programming a New Class of AI Accelerators (incl. Live Coding)](https://www.wearedevelopers.com/videos/100196-photonic-computing-programming-a-new-class-of-ai-accelerators-incl-live-coding) - [Geometric deep learning for drug discovery](https://www.wearedevelopers.com/videos/264-geometric-deep-learning-for-drug-discovery) - [Enhancing Workload Security in Kubernetes](https://www.wearedevelopers.com/videos/356-enhancing-workload-security-in-kubernetes) ## Related Articles - [Best US AI Conferences for CTOs in 2026: Build vs. Buy, Vendor Evaluation, and Peer Intelligence](https://www.wearedevelopers.com/magazine/736-best-us-ai-conferences-for-ctos-in-2026-build-vs-buy-vendor-evaluation-and-peer-intelligence) - [Dev Digest 121 - AI goes offline](https://www.wearedevelopers.com/magazine/456-dev-digest-121-ai-goes-offline) - [Dev Digest 120 - Apple and peers](https://www.wearedevelopers.com/magazine/455-dev-digest-120-apple-and-peers) - [What Are Large Language Models?](https://www.wearedevelopers.com/magazine/304-what-are-large-language-models) - [MLOps And AI Driven Development](https://www.wearedevelopers.com/magazine/82-mlops-and-ai-driven-development) - [Dev Digest 157: CUDA in Python, Gemini Code Assist and Back-dooring LLMs](https://www.wearedevelopers.com/magazine/557-dev-digest-157-cuda-in-python-gemini-code-assist-and-back-dooring-llms)