Software Engineer, Compute 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, 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

Build the compute infrastructure that supports Chai’s next generation of models, scientific products, and autonomous research. The capacity we bring online and how well we use it determine the scale of research we can pursue and the products we can put in scientists' hands.

• Bring new clusters online and build shared infrastructure for research, inference, and engineering.

• Work directly with researchers, engineers, and scientists to make that compute reliable and easy to use as the fleet and its demands grow.

• Build the shared storage and security foundations that let people and agents use compute and data safely.

• Optimize model inference and serving to run more scientific work with the compute we have.

About you

We look for depth in distributed systems, compute infrastructure, or performance engineering, with demonstrated ownership of substantial production systems. You can reason across layers, make sound tradeoffs around reliability and efficiency, and simplify infrastructure for the people using it.

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.