Chai Discovery
Research Engineer, ML Infrastructure
- Location
- San Francisco, CA, US
- Employment type
- Full-time
- Posted
- 24 September 2026 (about a month ago)
Checked 23 days agoApplications go to the employer, never to RoleSprint
About this role
ABOUT CHAI DISCOVERY
Chai Discovery builds the computer-aided design suite for molecules. Eli Lilly https://endpoints.news/eli-lilly-chai-discovery-sign-ai-software-deal/, GSK https://endpoints.news/chai-discovery-signs-gsk-deal-to-make-biologics-with-ai/, 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/, Bristol Myers Squibb https://www.linkedin.com/feed/update/urn:li:activity:7496300911523246080/, argenx https://x.com/chaidiscovery/status/2077422480617267230?s=20 and Novartis https://www.chaidiscovery.com/news/novartis-partnership all use Chai to power drug discovery programs. AI is reinventing life sciences the same way it reinvented software engineering, and our team is building that future.
We train frontier models that learn the underlying foundations of biochemical structure and interaction. We pair this with a computational workspace to iterate on molecules from idea to medicine, helping scientists move faster and develop medicines beyond the reach of existing methods.
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.
- Make our ML training stack maximally reliable: fault tolerance, checkpointing, and deterministic orchestration for long-running, large-scale jobs.
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 rigorous and craft-obsessed team.
ABOUT YOU
Ideal backgrounds include deep industry experience working with top AI/ML teams on the kinds of problems and systems we describe above—with strong software system design skills, proficiency in Python, and Pytorch or JAX fluency. We look for technical spikes where you have gone deep and demonstrated exceptional impact on real-world problems and systems.
WE OFFER
The opportunity to work at the vanguard of AI research and frontier biology, with world-class people, on a mission that matters. We protect and promote a culture of high velocity and ownership. We compensate our team accordingly.
Work location
- San Francisco, CA, US
Related jobs
Ready to make a decision?
This role is either worth your time or it isn’t.
Analyze the posting against your experience, see the gaps clearly, and build the right materials only if the opportunity makes sense.
Nothing is submitted automatically. You choose what happens next.
About this listing
Published on Ashby under the board identifier Chaidiscovery, which is the name the employer’s own job board carries. RoleSprint has not verified the company’s registered or trading name, so it is shown exactly as published rather than tidied up.
RoleSprint is not the employer and not a recruiter. Applications are made on the employer’s own site and never reach us; what RoleSprint does is help you decide whether a role is worth your time and prepare for it if it is.
Published 24 September 2026, last checked 23 days ago. A posting stops being advertised here 90 days after the employer published it, and one the employer takes down is marked closed rather than quietly removed.