Apply on Employer Site

Adaptive ML · 5 months ago

GPU Performance Engineer

New York / Paris

Full-time

Onsite

Mid Level

Adaptive ML is developing a reinforcement learning platform for specialized language models and is seeking a GPU Performance Engineer to enhance the performance of their LLM stack. The role involves building and maintaining GPU code, optimizing performance for various workloads, and contributing to the product roadmap.

A/B TestingArtificial Intelligence (AI)Information TechnologyMachine Learning

H1B Sponsored

Responsibilities

Build and maintain fast and robust GPU code, focusing on delivering performance improvements in real world applications

Write high-quality software in CUDA, CUTLASS, or Triton with a focus on performance and robustness

Profile dedicated GPU kernels, optimizing across latency/compute-bound regimes for complex workloads

Contribute to our product roadmap, by identifying promising trends that can improve performance

Report clearly on your work to a distributed collaborative team, with a bias for asynchronous written communication

Qualification

CUDAGPU optimizationPerformance tuningCUTLASSTritonSoftware engineeringAsynchronous communicationGenerative AI passion

Required

Build and maintain fast and robust GPU code, focusing on delivering performance improvements in real world applications

Write high-quality software in CUDA, CUTLASS, or Triton with a focus on performance and robustness

Profile dedicated GPU kernels, optimizing across latency/compute-bound regimes for complex workloads

Contribute to our product roadmap, by identifying promising trends that can improve performance

Report clearly on your work to a distributed collaborative team, with a bias for asynchronous written communication

A M.Sc. /Ph.D. in computer science, or demonstrated experience in software engineering, preferably with a focus on GPU-optimization

Strong programming skills, preferably with a focus on systems and general purpose GPU programming

A track record of writing high performance kernels, having preferably demonstrated ability to reach state of the art performance on well defined tasks

Contributions to relevant open-source projects, such as CUTLASS, Triton and MLIR

Passionate about the future of generative AI, and eager to build foundational technology to help machines deliver more singular experiences