- Employment
- Full-time
About the role
About Chai Discovery
Chai is a research lab working on AI to unlock biology. Our models design new molecules for new medicines. We are changing how biologists develop drugs, just as language models are changing how engineers write code. Our vision is a design suite for molecules, with applications across life sciences, agriculture, materials and beyond.
Our founders have been at the forefront of this field from the beginning. We are backed by Thrive, General Catalyst, OpenAI, Dimension and other tier-one investors. We partner with global life sciences companies like Eli Lilly on deals that are transforming industry.
We are known for talent density, rigorous research and pace of execution.
About the role
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. Chai's models are moving beyond protein structure prediction into real-world therapeutic engineering. 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.
Integrate AI models into production systems, ensuring scalability and reliability.
About you
Ideal backgrounds include high-performance computing, custom CUDA kernels and GPU programming, AI/ML + data systems at scale.
Bachelor’s or Master’s degree in Computer Science, Software Engineering, or a related field.
Strong proficiency in programming languages such as Python and deep learning frameworks like PyTorch.
Deep understanding of operating system internals.
Experience with HPC infrastructure such as Slurm and Kubernetes.
Experience with performance engineering and profiling for CPU and CPU workloads.
Experience writing, building, and maintaining large-scale ETL pipelines.
We offer
The opportunity to work at the frontier of AI, with world-class people, on a mission that matters. We protect & promote a culture of high velocity and ownership. We offer highly competitive compensation.
741,000+ hidden jobs like this
Chai Discovery and thousands of companies post here first — often days before LinkedIn or Indeed. Your first 5 applications are free; go Pro to apply without limits.
Everything Pro unlocks:
- Unlimited applications — free stops at 5
- Track every application in one place
- Apply straight to the source, one click
- Save & organize roles you love
- Roles pulled from company boards before the big sites
