Research Scientist / Engineer – Performance Optimization jobs in United States
cer-icon
Apply on Employer Site
company-logo

Luma AI · 3 days ago

Research Scientist / Engineer – Performance Optimization

positionPalo Alto, CA
timeFull-time
remoteOnsite
senioritySenior Level
money$180K/yr - $250K/yr

Luma AI is focused on enhancing the performance of their AI models through their Performance Optimization team. The role involves profiling and optimizing code for GPUs and CPUs, developing high-performance implementations, and researching cutting-edge optimization techniques for transformer models.

Artificial Intelligence (AI)Generative AIVideoVideo Editing
check
H1B Sponsor Likelynote

Responsibilities

Profile and optimize GPU/CPU/Accelerator code for maximum utilization and minimal latency
Write high-performance PyTorch, Triton, CUDA, deferring to custom PyTorch operations if necessary
Develop fused kernels and leverage tensor cores and modern hardware features for optimal hardware utilization on different hardware 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 model

Qualification

Triton programmingCUDA programmingGPU optimizationPyTorchProfiling toolsTransformer architecturesKernel developmentCustom operationsCompiler experienceInference optimizationKernel fusion techniquesWarp-level intrinsics

Required

Expert-level proficiency in Triton/CUDA programming and GPU optimization
Strong PyTorch skills
Experience with PyTorch 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/exporters such as torch.compile, TensorRT, ONNX, XLA
Experience optimizing inference workloads for latency and throughput
Experience with Triton compiler and kernel fusion techniques
Knowledge of warp-level intrinsics and advanced CUDA optimization

Company

Luma AI

twittertwittertwitter
company-logo
Luma AI develops tools that let users generate photorealistic images and videos from text, image, or video prompts.
dateFounded in 2021
location
Palo Alto, California, USA
people-size11-50 employees
web-link
https://lumalabs.ai

H1B Sponsorship

Luma AI has a track record of offering H1B sponsorships. Please note that this does not guarantee sponsorship for this specific role. Below presents additional info for your reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (10)
2024 (3)

Funding

Current Stage
Growth Stage
Total Funding
$1.06B
Key Investors
HUMAINAndreessen HorowitzAmplify Partners
2025-11-19Series C· $900M
2024-12-06Series B· $90M
2024-01-09Series B· $43M

Leadership Team

leader-logo
Amit Jain
Co-Founder
linkedin

Recent News

Morningstar.com
As AMD chases Nvidia, here's how it's positioning itself for the future of AI
2026-01-06
Business Wire
Luma AI Announces Ray3 Modify, a New Model for Hybrid-AI Workflows for Acting & Performances, Now Available in Dream Machine
2025-12-18
Crunchbase News
Jeff Bezos’s Project Prometheus Joins The Unicorn Board Alongside 18 Other Startups In November
2025-12-10
Company data provided by crunchbase