Research Scientist Engineer Performance Optimization
Luma AI develops multimodal creative AI agents, video and image generation models, and an API for generative media workflows.
Funding history
About Luma AI
Luma AI is an active AI company whose Luma Agents product plans, generates, iterates, and refines creative work across video, image, audio, and text. Its first-party materials describe proprietary Ray video and Uni image models, as well as developer API access.
Skills
Candidate Availability
Required and preferred rules are kept separate and reflect the wording in the original posting.
About the Role
You will profile and optimize GPU, CPU, and accelerator code for utilization and latency. You will write high-performance PyTorch, Triton, and CUDA, develop fused kernels, optimize distributed production deployments, build performance tooling, and implement transformer optimization techniques.
Requirements
- Expert-level Triton and CUDA programming
- GPU optimization expertise
- Strong PyTorch skills including kernel development and custom operations
- Proficiency with NVIDIA Nsight, torch profiler, or custom profiling tools
- Deep understanding of transformer architectures and attention mechanisms
Responsibilities
- Profile and optimize GPU, CPU, and accelerator code
- Write high-performance PyTorch, Triton, and CUDA code
- Develop fused kernels and use tensor cores
- Optimize model implementations for distributed multi-node deployment
- Build performance monitoring and analysis tools
- Research and implement transformer optimization techniques