Chai Discovery
Research Engineer
San Francisco office · On-site · Full-time · Posted Sep 5, 2024
- Experience
- Deep industry experience is described as an ideal background; no number of years stated.
About the job
Develop core frameworks that make Chai Discovery’s machine-learning models performant, fast, and reliable at scale, supporting AI-driven drug design.
What you'll do
- Build datasets and algorithms to evaluate and monitor model performance, and analyze failure modes with researchers.
- Own the distributed training stack, addressing performance and reliability bottlenecks across parallelism, quantization, and kernel optimization.
- Own and contribute to auto research and auto kernels frameworks.
What they ask for
- Strong software system design skills.
- Proficiency in Python.
- Fluency with Pytorch or JAX.
Nice to have
- Deep industry experience working with top AI/ML teams on relevant problems.
- Demonstrated exceptional impact on real-world problems and systems, with deep technical expertise.
Tools
Python, Pytorch, JAX
AI focus
Develop and optimize machine-learning training and evaluation systems for AI drug design, including models advancing from protein structure prediction to therapeutic engineering.
Model-written summary · the original posting is authoritative