Official Python implementation for the paper accepted at NeurIPS AI4Science 2025 'Beyond Ensembles: Simulating All-Atom Protein Dynamics in a Learned Latent Space'.