Member of Technical Staff (AI Inference Engineer)
Perplexity · London
6 months ago
Not specifiedFull_timellm inferencegpu performance optimizationdistributed systems
6 months ago
Workplace
Not specified
Employment
Full_time
Location
London
Posted
6 months ago
Summary
Join an ai inference engineering team building and operating the inference engine behind perplexity queries. You will support transformer retrieval, text-generation, and multimodal models, migrate gpu kernels to nvidia cute dsl, and develop a rust-native inference server for low-latency, cost-aware serving.
Responsibilities
- Support new model architectures in inference
- Manage weight loading and request scheduling
- Implement kv-cache management for inference
- Port cuda kernels to nvidia cute dsl
- Develop a rust-based inference serving runtime
- Profile and fix performance bottlenecks
- Build observability dashboards and incident remediation
rustpythoncudacute dslgpu programmingtransformer-based retrievaltext-generationmultimodal modelskv-cache managementapi gatewaygpu kernelsnvidia cute dslgpu performance optimizationprofilingcontinuous batchinggpu kernels interleavingllm architecturesproduction distributed systemstritoncutlasspytorchtorch.compilecustom operatorsncclnvlinkinfinibandrdmaint8fp8fp4quantizationmixed-precision servingnsight computensight systemscuda-gdbptx/sass analysiskubernetesgpu schedulingautoscalingjaxtensorflowquantizationspeculative decodingprefill-decode disaggregation
llm inferencegpu performance optimizationdistributed systemsmodel serving infrastructurequantization and low-precision inference
