ChaiDiscovery logoChaiDiscovery logo

AI Research Engineer at Chai Discovery | San Francisco

ChaiDiscoverySan Francisco office
On-site Full-time

Clicking Apply Now takes you to AutoApply where you can tailor your resume and apply.


Experience Level

Experience

Qualifications

Bachelor’s or Master’s degree in Computer Science, Software Engineering, or a related field. Strong proficiency in programming languages such as Python and deep learning frameworks like PyTorch. Deep understanding of operating system internals. Experience with HPC infrastructure such as Slurm and Kubernetes. Experience with performance engineering and profiling for CPU and GPU workloads. Experience writing, building, and maintaining large-scale ETL pipelines.

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.

About ChaiDiscovery

Chai Discovery is at the forefront of transforming drug discovery through innovative AI models and techniques. Our team is composed of leading experts who are committed to re-engineering the therapeutic development process to improve patient outcomes.

Similar jobs

Browse all companies, explore by city & role, or SEO search pages. View directory listings: all jobs, search results, location & role pages.

Tailoring 0 resumes

We'll move completed jobs to Ready to Apply automatically.