[NeurIPS'2024] Exploring Behavior-Relevant and Disentangled Neural Dynamics with Generative Diffusion Models Official Repo