AI Research Engineer

Chai Discovery • San Francisco, CA

9d • Onsite

About The Position

About Chai Discovery Chai Discovery builds frontier AI models to design molecules and transform how new cures are found. We are a craft-obsessed team setting out to re-architect drug discovery and save lives. The Chai founding team brought together leading researchers in this space and top silicon valley operators. Our founders led seminal accomplishments at top AI-for-biology programs and labs, co-invented protein language modelling, built state-of-the-art folding algorithms, and sold AI adopted by top-10 pharma companies. We are backed by top-tier investors, including OpenAI, Thrive Capital, Dimension, Conviction, Lachy Groom, Amplify, and many more. 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.

Requirements

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

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.

Benefits

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

What This Job Offers

• Job Type: Full-time
• Career Level: Mid Level
• Number of Employees: 1-10 employees