Research Scientist / Engineer, Luma AI
Performance Optimization
$180-250k
+ Equity in the form of stock options
3D AI design platform
Job no longer available
3D AI design platform
21-100 employees
Job no longer available
$180-250k
+ Equity in the form of stock options
21-100 employees
Company mission
To enable everyone to capture and experience the world in lifelike 3D.
Role
Who you are
- 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
- (Preferred) Experience optimizing inference workloads for latency and throughput
- (Preferred) Experience with Triton compiler and kernel fusion techniques
- (Preferred) Knowledge of warp-level intrinsics and advanced CUDA optimization
- (Preferred) Background in compiler optimization or hardware-software co-design
What the job involves
- The Performance Optimization team at Luma is dedicated to maximizing the efficiency and performance of our AI models
- Working closely with both research and engineering teams, this group ensures that our cutting-edge multimodal models can be trained efficiently and deployed at scale while maintaining the highest quality standards
- 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
Application process
- Your applications are reviewed by real people
Our take
Luma AI is an AI-based API platform that harnesses the power of NeRF (neural radiance field) technology to create 3D models of any object through its mobile application. The generated 3D models can then be used for a range of digital uses, changing how people share memories, search for products, and reimagining internet and gaming spaces.
By utilizing NeRF technology, the company can bypass traditional programming requirements such as teams of 3D artists or expensive photogrammetry setups, thus democratizing 3D creation by making it accessible to anyone with the app.
While Luma AI faces a growing number of competitors in the space, it will need to focus on advancing and refining its technology to fully compete with AI giants such as OpenAI. However, the company’s focus on its specific area of the market certainly allows it to see vast growth opportunities.
Freddie
Company Specialist at Welcome to the Jungle
Company
Funding (last 2 of 4 rounds)
Dec 2024
$20m
LATE VC
Jan 2024
$43m
SERIES B
Total funding: $87.3m
Company benefits
- Equity grant to reflect the incredible value you will bring to Luma, with annual refreshes
- Full health, dental, and vision coverage
- Latest and greatest gear
- Stipends towards wellness, house cleaner, and phone/internet
- Unlimited paid time off with 12 days minimum
- Unlimited sick days
Company HQ
West Oakland, Oakland, CA
Leadership
Amit Jain
(Co-Founder)Previously worked at Apple as a Computer Vision, Cameras, and Systems Engineer. Founder and former Lead Developer of Cultured Pixel.
Alex Y.
(Co-Founder)Initially served as a Research Scientist at the company. Former Research Assistant at UC Berkeley. Also held Internships at Adobe and Google.
Share this job