GrepJob
Perplexity

UK Internship Program

Perplexity
Apply
5 months ago
London, United KingdomIntern
H1B Sponsor

Responsibilities

  • Work with the inference team to improve serving latency and throughput.
  • Support new models and state-of-the-art inference optimizations or quantization schemes.
  • Optimize inference across the entire stack, from GPU kernels to serving endpoints.

Requirements

  • Strong engineering track record with knowledge of programming fundamentals.
  • Pursuing a Master's or PhD in Computer Science focused on performance-related subjects.
  • Experience with ML frameworks like Torch or JAX.
  • Experience with GPU programming using CUDA or Triton.
  • Experience with High-Performance Computing (OpenMPI).

Categories

AI & MLData Engineering