Salary Range: 180000 to 250000 (Currency: USD) (Pay period: per-year-salary)

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.