AI Research Engineer

About Chai Discovery

Position Overview

Key Responsibilities

• Collaborate with AI researchers to design, implement, and maintain software systems that support AI-driven drug discovery applications.
• Develop and optimize high-performance computing (HPC) infrastructure.
• Integrate AI models into production systems, platforms, and front-end applications, ensuring scalability and reliability.

Qualifications

• Bachelor’s or Master’s degree in Computer Science, Software Engineering, or a related field.
• Strong proficiency in programming languages such as Python .
• Deep understanding of operating system internals (Unix).
• Experience with high-performance computing (HPC) infrastructure such as Slurm and Kubernetes.
• Experience with performance engineering (familiarity with profiling tools, optimizing CPU and GPU workloads, writing CUDA and Triton kernels, and so forth)
• Experience writing, building, and maintaining large-scale ETL pipelines.
• Experience using deep learning libraries such as PyTorch.

What We Offer

• Highly engaging work at the frontier of AI-driven drug discovery that will fundamentally change how new drugs are created.
• Opportunity to work with world-renowned leaders in the AI x biology ecosystem, contributing to impactful advancements in the field.
• Highly competitive salary and equity package.
• Comprehensive healthcare and wellness benefits.
• Continuous learning and career growth opportunities.
• Collaborative and supportive startup culture.
• A high-energy team committed to increasing the efficiency of creating new medicines and positively impacting millions of lives.
• Chai Discovery, Inc. is an equal opportunity employer. We encourage applications from candidates of all backgrounds and experiences to apply.