Karve
Browse all jobs

Member of Technical Staff (AI Inference Engineer)

Perplexity · London

6 months ago

Not specifiedFull_timellm inferencegpu performance optimizationdistributed systems

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