Research Engineer - AI-Driven Drug Discovery Systems

Chai Discovery Inc.; San Francisco (CA); On-site; USD 180,000 - 240,000; Full time

Job summary

Chai Discovery Inc. in San Francisco builds agent-powered systems to accelerate AI-driven drug discovery. This role focuses on designing frameworks that speed up model training and inference, enabling the research team to move faster and explore new targets.

Ideal candidates have deep ML experience, strong system design skills, and fluency in Python with PyTorch or JAX. You will work with a rigorous, craft-obsessed team at the vanguard of AI and biology.

Qualifications

• Focus on building agent-powered software systems for AI research.
• Strong experience in Python-based ML frameworks and tooling.
• Proven ability to design robust, scalable software systems for research workflows.

Responsibilities

• Build agent-powered systems and frameworks to accelerate model training and inference.
• Collaborate with research teams to identify bottlenecks and propose scalable solutions.
• Contribute to tooling and datasets that enable rapid experimentation and iteration.

Skills

Python

Pytorch

JAX

System design

Job description

Chai Discovery Inc. in San Francisco builds agent-powered systems to accelerate AI-driven drug discovery. This role focuses on designing frameworks that speed up model training and inference, enabling the research team to move faster and explore new targets.

Ideal candidates have deep ML experience, strong system design skills, and fluency in Python with PyTorch or JAX. You will work with a rigorous, craft-obsessed team at the vanguard of AI and biology.