Add the simple diffusion model tutorial (#83)

Co-authored-by: Melissa Weber Mendonça <melissawm@gmail.com>

Author: lucianommartins

.pre-commit-config.yaml

@@ -1,6 +1,6 @@
 repos:
   - repo: https://github.com/pre-commit/pre-commit-hooks
-    rev: v4.6.0
+    rev: v5.0.0
     hooks:
       - id: check-ast
       - id: check-merge-conflict

docs/conf.py

@@ -38,6 +38,7 @@
     'getting_started_with_jax_for_AI.md',
     'JAX_for_PyTorch_users.md',
     'JAX_porting_PyTorch_model.md',
+    'digits_diffusion_model.md',
     'JAX_for_LLM_pretraining.md',
 ]
 
@@ -64,5 +65,6 @@
 nb_execution_excludepatterns = [
     'JAX_for_PyTorch_users.ipynb',
     'JAX_porting_PyTorch_model.ipynb',
+    'digits_diffusion_model.ipynb',
     'JAX_for_LLM_pretraining.ipynb',
 ]