Research Scientist / Engineer – Performance Optimization

Redwood City, CA · Remotefull time

Apply on Employer Site

Remote

$195k–$395k

H1B: yes

Luma AI Research Scientist / Engineer – Performance Optimization at Luma AI focused on profiling and optimizing GPU, CPU, and accelerator code for efficient training and deployment of multimodal models. Develop fused kernels and leverage modern hardware features for distributed multi-node production deployment.

Responsibilities

Profile and optimize GPU/CPU/accelerator code for maximum utilization and minimal latency
Write high-performance PyTorch, Triton, and CUDA code, including custom operations
Develop fused kernels and leverage tensor cores and modern hardware features
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

Qualification

Strong PyTorch skillsAbout Luma

Required

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

Preferred

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.

Benefits

Vision

Company

Luma AI

View company page

Source: ashby

Active within the past hour

Remote-friendly

More roles at Luma AI

Editor + Motion Graphics

Los Angeles, CA · Remote

Software Engineer, Inference

Redwood City, CA · Remote

Research Scientist / Engineer – Training Infrastructure

Redwood City, CA · Remote