Full Job Description
Build agent-powered systems and frameworks to improve model training, accelerate inference, and help our research organization move faster. Automate the research loop itself rather than just the models it produces, including the tooling and datasets that allow agents to run experiments, evaluate results and propose next steps.
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.
Ideal backgrounds include deep industry experience working with top AI/ML teams on agent-powered software-with strong 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.
