Chai Discovery

Research Engineer

San Francisco office

Sponsorship not specifiedDetected 686 days ago

PythonMachine LearningDeep LearningData EngineeringResearch

About the role

Chai's models are moving beyond protein structure prediction into real-world therapeutic engineering.
Integrate AI models into production systems, ensuring scalability and reliability.
Ideal backgrounds include high-performance computing, custom CUDA kernels and GPU programming, AI/ML + data systems at scale.

Chai is a research lab working on generative models for molecular design.
This is a chance to push the frontier of AI drug design, working alongside a craft-obsessed team of micro-pessimists and macro-optimists.
Develop and optimize all the infrastructure & tooling upon which our AI research depends.
Collaborate with AI researchers to understand, implement and optimize new research directions.

Bachelor's or Master's degree in Computer Science, Software Engineering, or a related field.
Experience with HPC infrastructure such as Slurm and Kubernetes.
Experience with performance engineering and profiling for CPU and CPU workloads.

We are building AI infrastructure to engineer new medicines with speed and precision.
Our mission is to unlock progress towards better cures and better science, and we see countless interesting problems on the road ahead.
We are known for talent density, rigorous research and pace of execution.
We are backed by Sequoia, Index, Thrive, General Catalyst, Dimension, OpenAI and others.
We are seeking an AI Research Engineer to help design, train, evaluate, and optimize Chai's core models and infrastructure, working in tight loops with research scientists.
We offer highly competitive compensation.

This listing is sourced directly from Chai Discovery's careers page and normalized into a canonical job model.