Software Engineer, Compute Infrastructure

About Chai Discovery

Chai Discovery builds the computer-aided design suite for molecules. Eli Lilly, GSK, Pfizer, Bristol Myers Squibb, argenx and Novartis 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

Build the compute infrastructure behind Chai’s next generation of models, scientific products, and autonomous research. Our research program and runtime inference constantly demand more compute, and the rate at which we bring it online and manage it effectively directly inflects Chai's velocity.

• Bring new clusters up and own the systems that make them useful across the business.
• Work with researchers, engineers, and scientists to remove performance and reliability bottlenecks across the stack.
• Optimize model inference and serving to run more scientific work with available compute.
• Build the shared storage and security foundations that let people and agents use compute and data safely.

About you

We look for depth in distributed systems, compute infrastructure, and performance engineering, with demonstrated ownership of substantial production systems. You can reason across layers and make sound tradeoffs between complexity, reliability, and efficiency.

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.