Senior Research Engineer, Luma AI

• We are looking for engineers with significant experience solving hard problems in PyTorch, CUDA and distributed systems
• Experience training large models using Python & Pytorch, including practical experience working with the full development pipeline from data processing, preparation & data loading to training and inference
• Experience optimizing and deploying inference workloads for throughput and latency across the stack (inputs, model inference, outputs, parallel processing etc.)
• Experience with profiling CPU & GPU code in PyTorch, including Nvidia Nsight or similar
• Experience writing & improving highly parallel & distributed PyTorch code, with familiarity in DDP, FSDP, Tensor Parallel, etc
• Experience writing high-performance parallel C++. Bonus if done within an ML context with PyTorch, like for data loading, data processing, inference code
• Experience with high-performance Triton / CUDA and writing custom PyTorch kernels. Top candidates will be able to utilize tensor cores; optimize performance with CUDA memory and other similar skills
• Good to have experience working with Deep Learning concepts such as Transformers & Multimodal Generative models such as Diffusion Models and GANs
• Good to have experience building inference / demo prototype code (incl. Gradio, Docker etc.)

• You will work with Research Scientists to build & train cutting edge foundation models on thousands of GPUs
• Ensure efficient implementation of models & systems for data processing, training, inference and deployment
• Identify and implement optimization techniques for massively parallel and distributed systems
• Identify and remedy efficiency bottlenecks (memory, speed, utilization) by profiling and implementing high-performance CUDA, Triton, C++ and PyTorch code
• Work closely together with the research team to ensure systems are planned to be as efficient as possible from start to finish
• Build tools to visualize, evaluate and filter datasets
• Implement cutting-edge product prototypes based on multimodal generative AI

• 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

Total funding: $87.3m

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.