Copy of Research Scientist / Engineer – Performance Optimization

Luma; Greater London; Sur place; GBP 120 000 - 180 000; Plein temps

Résumé du poste

Luma is seeking a senior performance engineer to accelerate its multimodal models. You will profile and optimize GPU, CPU, and accelerator code to train faster and deploy at scale without compromising quality.

You will write kernels and operations that extract maximum throughput from modern hardware. This is deep performance work requiring expertise in CUDA/Triton, transformer internals, and distributed multi-node deployment.

Qualifications

• Expert-level Triton/CUDA programming and GPU optimization.
• Strong PyTorch skills, including kernel development and custom operations.
• Proficiency with profiling tools (NVIDIA Nsight, torch profiler, custom tooling).
• Deep understanding of transformer architectures and attention mechanisms.

Responsabilités

• Profile and optimize GPU/CPU/accelerator code for maximum utilization and minimal latency.
• Write high-performance PyTorch, Triton, and CUDA, dropping to custom operations when needed.
• Develop fused kernels and leverage tensor cores and modern hardware features across platforms.
• Optimize model architectures and implementations for distributed multi-node production deployment.
• Build performance monitoring and analysis tools and automation.
• Research and implement cutting-edge optimization techniques for transformer models.

Connaissances

Triton/CUDA

GPU optimization

PyTorch kernel development

Profiling tools

Transformer architectures

Outils

Nsight

Torch Profiler

TensorRT

Description du poste

You'll make Luma's multimodal models fast - profiling and optimizing GPU, CPU, and accelerator code so they train efficiently and deploy at scale without sacrificing quality. You'll write the kernels and operations that get the most out of the hardware.

This is deep performance work: fused kernels, tensor cores, Triton and CUDA, distributed multi-node deployment. It fits someone with expert GPU-optimization skills and a deep understanding of transformer internals. If you're not at home in CUDA, Triton, and profilers, this is the wrong depth.

What You'll Own

• Profile and optimize GPU/CPU/accelerator code for maximum utilization and minimal latency.
• Write high-performance PyTorch, Triton, and CUDA, dropping to custom operations when needed.
• Develop fused kernels and leverage tensor cores and modern hardware features across platforms.
• Optimize model architectures and implementations for distributed multi-node production deployment.
• Build performance monitoring and analysis tools and automation.
• Research and implement cutting-edge optimization techniques for transformer models.

First 90 Days

One way the first 90 could unfold.

• Days 1-30 - Immerse & Diagnose: Profile the current training and inference paths and find the biggest performance wins.
• Days 30-60 - Ship & Validate: Land a kernel or architecture optimization that measurably improves utilization or latency.
• Days 60-90 - Scale & Systemize: Build the monitoring and automation that keeps performance gains from regressing.

What You Bring

• Expert-level Triton/CUDA programming and GPU optimization.
• Strong PyTorch skills, including kernel development and custom operations.
• Proficiency with profiling tools (NVIDIA Nsight, torch profiler, custom tooling).
• Deep understanding of transformer architectures and attention mechanisms.

Nice to Have

• Experience with compilers and exporters (torch.compile, TensorRT, ONNX, XLA).
• Experience optimizing inference workloads for latency and throughput.
• Triton compiler and kernel fusion techniques.
• Knowledge of warp-level intrinsics and advanced CUDA optimization.

About Luma: Luma's mission is to build unified general intelligence that can generate, understand, and operate in the physical world. We believe multimodality is critical for intelligence - the next step beyond language models comes from vision. Luma is an equal opportunity employer.