Research Engineer, Luma AI

Performance Optimization

$180-250k

+ Equity in the form of stock options

Docker

Python

Tensorflow

C++

CUDA

PyTorch

Mid and Senior level

San Francisco Bay Area

Office located in Palo Alto, CA

3D AI design platform

Be an early applicant

3D AI design platform

21-100 employees

B2BArtificial IntelligenceAugmented RealitySaaSAPI

Be an early applicant

$180-250k

+ Equity in the form of stock options

Docker

Python

Tensorflow

C++

CUDA

PyTorch

Mid and Senior level

San Francisco Bay Area

Office located in Palo Alto, CA

21-100 employees

B2BArtificial IntelligenceAugmented RealitySaaSAPI

Company mission

To enable everyone to capture and experience the world in lifelike 3D.

Job

Company

Role

Who you are

We are looking for engineers with significant problem solving experience in PyTorch, CUDA and distributed systems
Experience training large models using Python & Pytorch, including practical experience working with the entire 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.)

What the job involves

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

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.

Salary benchmarks

We don't have enough data yet to provide salary benchmarks for this role.

Submit your salary to help other candidates with crowdsourced salary estimates.

Share this job

View 29 more jobs at Luma AI