Research Engineer - ML Infrastructure

Chai Discovery seeks a Research Engineer to optimize machine learning infrastructure for molecular design models. The role involves building training systems, profiling GPU clusters, and ensuring scalability and reliability for large-scale computational workloads supporting drug discovery. Responsibilities Develop and optimize training stack across model, layer, and kernel levels Profile end-to-end training on large GPU clusters; identify and eliminate bottlenecks Ensure new model architectures…