Job Details

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.

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.

• 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.

• 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.