About the job
About Chai Discovery
At Chai Discovery, we are pioneering advanced AI models that revolutionize molecular design and the drug discovery process. Our dedicated team is committed to reshaping the landscape of therapeutic development, with the ultimate goal of saving lives.
Our founders consist of elite researchers and industry leaders who have made groundbreaking contributions to AI applications in biology. They have co-invented protein language modeling, developed cutting-edge folding algorithms, and created AI solutions adopted by top pharmaceutical firms. We are proud to be backed by prominent investors, including OpenAI, Thrive Capital, Dimension, Conviction, Lachy Groom, and Amplify.
About the Role
We are looking for an AI Research Engineer to collaborate closely with our research scientists in designing, training, evaluating, and optimizing the core models and infrastructure at Chai. Our models are evolving from mere protein structure predictions into real-world therapeutic engineering applications. This is a unique opportunity to advance the field of AI-driven drug design while being part of a passionate team that balances cautious optimism with bold vision.
- Develop and enhance the infrastructure and tools that support our AI research initiatives.
- Work alongside AI researchers to explore, implement, and refine emerging research directions.
- Integrate AI models into production environments, ensuring they are scalable and reliable.
About You
The ideal candidate will have a background in high-performance computing, GPU programming, and AI/ML with large-scale data systems.
- A Bachelor’s or Master’s degree in Computer Science, Software Engineering, or a related discipline.
- Proficiency in programming languages such as Python, along with deep learning frameworks like PyTorch.
- A solid understanding of operating system internals.
- Familiarity with HPC infrastructure, including Slurm and Kubernetes.
- Experience in performance engineering and profiling for both CPU and GPU workloads.
- Proven ability in developing, building, and maintaining extensive ETL pipelines.
We Offer
- Innovative Technology: Engage with cutting-edge technologies that have the potential to redefine therapeutic creation.
- Exceptional Team: Collaborate with renowned experts at the intersection of AI and biology.
