← All roles

Research Engineer

Chai Discovery· Research
Apply on company site
San Francisco officeFull-timePosted 11mo ago

About the role

Make our models performant, fast, and reliable at scale by developing the core frameworks used to train and evaluate our ML models. Chai Discovery builds AI-driven drug design tools and trains frontier models for biochemical structure and interaction, pairing them with a computational workspace to move molecules from idea to medicine.

Responsibilities

  • Build datasets and algorithms to systematically evaluate and monitor model performance.
  • Analyze model failure modes, working closely with researchers on experiments to mitigate them.
  • Own the distributed training stack, removing runtime and reliability bottlenecks across all layers of the stack and architecture—parallelism, quantization, and kernel optimization and whatever else might move the needle.
  • Own and contribute to our auto research and auto kernels frameworks.

Qualifications

  • Ideal backgrounds include deep industry experience working with top AI/ML teams on the kinds of problems we describe above—with strong software system design skills, proficiency in Python, and Pytorch or JAX fluency.

Skills

Required:PythonPyTorch or JAX
Last updated Oct 14, 2026. Interested? Apply on the company site.

About Chai Discovery

Building frontier artificial intelligence to predict and reprogram the interactions between biochemical molecules.

Building frontier artificial intelligence to predict and reprogram the interactions between biochemical molecules.

artificial intelligencebiochemical moleculesliquid-ai-recruiting-target
Company profileWebsite

More open roles at Chai Discovery