Research Engineer

We are looking for 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 the performance of models 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 find inefficiencies and bottlenecks in model training, and data loading

Collaborate with cross-functional teams including data, applied research and infrastructure

Experience

Very strong demonstrated engineering ability in Python and Pytorch.

Experience building ML models from scratch in Pytorch.

Academic or Professional experience with (and understanding of) generative multimodal models such as Diffusion Models and GANs, as well as deep learning concepts such as Transformers.

Good to have familiarity with Linux clusters, systems & scripting.

Good to have experience working with large distributed systems (>100 GPUs).

Compensation

The pay range for this position in California is $180,000 - $250,000yr; however, base pay offered may vary depending on job-related knowledge, skills, candidate location, and experience. We also offer competitive equity packages in the form of stock options and a comprehensive benefits plan.