Chai 2 is described as an all-atom diffusion model.