Research Engineer, ML Infrastructure
Chai Discovery · San Francisco office
Bio & Science AIonsite
At a glance
Highlights
- Build core frameworks to make frontier biochemical models performant at scale.
- Optimize training via parallelism, quantization, and custom kernels on GPU clusters.
- Push beyond protein prediction into real-world therapeutic engineering.
What they offer
- Work at the vanguard of AI research and frontier biology on a mission that matters
- High-velocity, ownership-driven culture with world-class team
Ideal candidate
Deep systems builder who has driven exceptional impact on large-scale ML infrastructure and cares about reliable, efficient training to advance real-world therapeutic design.
Requirements
- Deep industry experience working with top AI/ML teams on large-scale systems
- Strong software system design skills
- Proficiency in Python
- PyTorch or JAX fluency
- Demonstrated exceptional impact on real-world problems and systems
Lab profile
About Chai Discovery
Builds an AI computer-aided design suite for molecules.
What they build
- Computer-aided molecular design suite
- AI biologics design tools
- Antibody design platform
Known for
- Molecular pre-training research
- Antibody engineering
- Pharma deployment
Key facts
- Collaborations with Eli Lilly, GSK and Pfizer
Posted · Last seen