AI Engineer

About Chai Discovery

Chai Discovery is building AI foundation models to enable drug discovery projects for biotech and pharma. The company is backed with significant capital by top-tier investors, such as OpenAI, Thrive Capital, Dimension, Conviction, Lachy Groom, Amplify, and many more.

The company recently came out of stealth by launching their first model, Chai-1, the world's most performant model for molecular structure prediction. For example, on the Posebusters benchmark, Chai-1 achieves a success rate of 77%, while AlphaFold-3 from Google DeepMind achieves 76%.

Chai is operated by a highly experienced founding team, who are well known in the AI community for seminal research accomplishments at top AI labs. The team have run AI departments at top AI x Biology organizations, have exited a past company via IPO, and sold AI technologies to multiple top 10 pharma companies.

Position Overview

Chai Discovery is seeking an AI Engineer to play a crucial role in developing our next-generation AI x biology foundation models and application to real-world drug discovery challenges. The AI Engineer will work closely with the founding team to build, optimize, and maintain the software and infrastructure that power our AI-driven drug discovery pipeline. This position is based full-time in San Francisco, CA.

Key Responsibilities

• Collaborate with AI researchers to design, implement, and maintain software systems that support AI-driven drug discovery applications.

• Develop and optimize high-performance computing (HPC) infrastructure.

• Integrate AI models into production systems, platforms, and front-end applications, ensuring scalability and reliability.
  

Qualifications

• Bachelor’s or Master’s degree in Computer Science, Software Engineering, or a related field.

• Strong proficiency in programming languages such as Python .

• Deep understanding of operating system internals (Unix). 

• Experience with high-performance computing (HPC) infrastructure such as Slurm and Kubernetes.

• Experience with performance engineering (familiarity with profiling tools, optimizing CPU and GPU workloads, writing CUDA and Triton kernels, and so forth)

• Experience writing, building, and maintaining large-scale ETL pipelines.

• Experience using deep learning libraries such as PyTorch.
  

What We Offer

• Highly engaging work at the frontier of AI-driven drug discovery that will fundamentally change how new drugs are created.

• Opportunity to work with world-renowned leaders in the AI x biology ecosystem, contributing to impactful advancements in the field.

• Highly competitive salary and equity package.

• Comprehensive healthcare and wellness benefits.

• Continuous learning and career growth opportunities.

• Collaborative and supportive startup culture.

• A high-energy team committed to increasing the efficiency of creating new medicines and positively impacting millions of lives.
  

Chai Discovery, Inc. is an equal opportunity employer. We encourage applications from candidates of all backgrounds and experiences to apply.