Research Engineer II
zebra
Research Engineer - ML Infrastructure
chaidiscovery· Research
About this role
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, Pfizer, and Novartis 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.
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 & promote a culture of high velocity and ownership. We compensate our team accordingly.
Frequently Asked Questions
Is the salary disclosed for the Research Engineer - ML Infrastructure position at chaidiscovery?
The salary for this Research Engineer - ML Infrastructure role at chaidiscovery is not publicly listed. Click "Apply Now" to learn more about the compensation package on their official careers page.
Where is the Research Engineer - ML Infrastructure position at chaidiscovery located?
This Research Engineer - ML Infrastructure role at chaidiscovery is based in San Francisco office. The position is listed as on-site or hybrid. Check the full job description or apply directly to confirm the work arrangement.
Is the Research Engineer - ML Infrastructure role at chaidiscovery full-time or part-time?
This is listed as a FullTime position. It is posted as a Research Engineer - ML Infrastructure role in the Research department at chaidiscovery.
Which team or department does the Research Engineer - ML Infrastructure at chaidiscovery belong to?
This Research Engineer - ML Infrastructure position is part of the Research department at chaidiscovery. See the full job description for more information about the team structure and responsibilities.
How do I apply for the Research Engineer - ML Infrastructure position at chaidiscovery?
Click the "Apply Now" button on this page. You will be redirected to chaidiscovery's official application portal hosted on ashby where you can submit your application directly.
When was the Research Engineer - ML Infrastructure job at chaidiscovery posted?
This Research Engineer - ML Infrastructure position at chaidiscovery was posted on Sep 24, 2026. Apply as soon as possible — early applications are often reviewed first.
Research Engineer - ML Infrastructure
chaidiscovery
You'll be redirected to chaidiscovery's official application page on Ashby ATS.