Senior Research Engineer - Foundation Models

Luma AI; Palo Alto (CA); On-site; USD 180,000 - 250,000; Full time

Job summary

A leading AI company seeks a research engineer to enhance its generative models. Join a dynamic team to develop cutting-edge algorithms and optimize performance across vast GPU clusters. In addition to a competitive salary, you will also receive equity options, ensuring your contribution to the company’s success is recognized.

Benefits

Stock options

Comprehensive benefits

Qualifications

• Strong engineering ability in Python and PyTorch.
• Experience with multimodal models like Diffusion Models and GANs preferred.
• Knowledge of large distributed systems and Linux is a plus.

Responsibilities

• Lead research in multimodal foundation models and algorithm development.
• Optimize AI model performance focusing on deployment efficiency.
• Collaborate with cross-functional teams on data and infrastructure.

Skills

Python

PyTorch

Generative AI

Deep Learning

Linux

Job description

We are looking for a research engineer with extremely strong technical experience in training Generative AI models. You’ll be part of the research team, helping build our core multimodal foundation models and manage training runs across thousands of GPUs.

Responsibilities

• Lead and contribute to cutting-edge research in multimodal foundation models
• Design, develop, and experiment with novel algorithms, architectures, and techniques that enhance the performance, efficiency, and scalability of our AI models
• Optimize model performance for deployment in production environments, focusing on latency, throughput, and computational efficiency without compromising accuracy or robustness
• Inspect and manage large-scale data clusters to identify inefficiencies and bottlenecks in model training and data loading
• Collaborate with cross-functional teams including data, applied research, and infrastructure

Experience

• Strong engineering ability in Python and PyTorch
• Experience building ML models from scratch in PyTorch
• Academic or professional experience with generative multimodal models such as Diffusion Models and GANs, and deep learning concepts like Transformers
• Familiarity with Linux clusters, systems, and scripting is a plus
• Experience working with large distributed systems (>100 GPUs) is a plus

Compensation

• The pay range for this position in California is $180,000 - $250,000 per year; actual base pay may vary based on knowledge, skills, location, and experience. We also offer competitive equity packages in the form of stock options and comprehensive benefits.

Additionally, the total compensation can be $250,000 - $300,000 per year, including a sizable equity grant.

In addition to cash base pay, you'll also receive a significant grant of Luma's equity. The pay range is for the Bay Area, and base pay may vary depending on candidate's qualifications and location.

Your applications are reviewed by real people.