This repository has been made for the computer session about the lectures done by prof. Leonardo Zepeda-Núñez.
It consists of three notebooks meant to be worked in the specified order with the following topics
- Jax tutorial
- Langevin Monte Carlo tutorial
- Introduction to Score-Based generative models (SGM)
The libraries required to run this notebook are
numpymatplotlibScipyJax(on cpu should be enough)Flax(Jax has to be installed before installing it)
Disclaimer: Part of the notebooks were based off of different existing notebooks/tutorials on Jax/Flax for diffusion models.