ML · Infrastructure
ABOUT CHAI DISCOVERY
Chai builds the design suite for molecules. We train frontier models that learn the underlying foundations of biochemical structure and interaction, so scientists can move faster and pursue targets that other methods cannot reach.
AI is reinventing life sciences the same way it reinvented software engineering, and Chai is at the forefront of this shift. Leading pharmaceutical companies like Eli Lilly https://endpoints.news/eli-lilly-chai-discovery-sign-ai-software-deal/, Pfizer https://www.forbes.com/sites/amyfeldman/2026/06/04/why-pfizer-and-eli-lilly-are-betting-on-this-13-billion-ai-drug-discovery-startup/, and Novartis https://www.chaidiscovery.com/news/novartis-partnership are adopting our platform to power their drug discovery programs.
We value diverse perspectives and are ready to find greatness in unexpected places.
ABOUT THE ROLE
Make our models performant, resource efficient and reliable at scale by developing the core frameworks for model training and evaluation, in close partnerships with fellow researchers and engineers.
- Build our training stack across model, layer, and kernel levels; optimize workloads through parallelism, quantization, and custom kernels.
- Profile end-to-end training runs on large GPU clusters; eliminate bottlenecks and failures; monitor throughput, utilization, and uptime.
- Ensure new model architectures and training recipes scale efficiently, from early experiments to frontier-scale runs.
- Own reliability of the training stack: fault tolerance, checkpointing, and deterministic orchestration for long-