AI Research Engineer (San Francisco)

About Chai Discovery

Chai Discovery is unlocking new biology with artificial intelligence by building state-of-the-art foundation models. The company is operating in stealth mode by a highly experienced founding team, who are well known in the AI community for seminal research accomplishments at top AI labs, 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. The company is backed with significant capital by top-tier investors.

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

• Bachelors or Masters 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.