Pedagogical codebase for a simplified score-based generative model design, with training loop - View it on GitHub