About this role
This Research Engineer role suits professionals with deep industry experience working with top artificial intelligence and machine learning teams on complex systems and problems. Based at the company office in San Francisco, the position involves developing core frameworks to make machine learning models performant, resource-efficient, and reliable at scale.
Role Overview
Chai Discovery seeks a Research Engineer specializing in machine learning infrastructure for its San Francisco office location. The successful candidate will collaborate closely with fellow engineers and researchers to build core frameworks for model training and evaluation.
Key Responsibilities
Requirements
Skills and Tools
About the Company
Chai Discovery builds a design suite for molecules by training frontier models that learn foundational biochemical structures and interactions. Pharmaceutical companies such as Eli Lilly, Pfizer, and Novartis utilize the platform for drug discovery programs. The organization values diverse perspectives and maintains a culture centered on high velocity and ownership.
How to Apply
Interested candidates should submit their application directly on the official company careers page.
Skills and tech stack
Questions candidates ask
Where is the Research Engineer position located?
The role is located at the company office in San Francisco.
What programming languages and frameworks are required?
The position requires proficiency in Python along with fluency in either PyTorch or JAX.
What are the primary responsibilities for this role?
Responsibilities include building training stacks across model, layer, and kernel levels, optimizing workloads, profiling training runs on large GPU clusters, scaling model architectures, and ensuring reliability through fault tolerance and checkpointing.
What kind of background is ideal for applicants?
Ideal candidates have deep industry experience working with top artificial intelligence and machine learning teams on complex systems, supported by strong software system design skills and a technical spike showing exceptional impact.