Add the simple diffusion model tutorial (#83)

Co-authored-by: Melissa Weber Mendonça <melissawm@gmail.com>
jax-ml · Nov 14, 2024 · 0197a4b · 0197a4b
1 parent 12a0bb8
commit 0197a4b
Show file tree

Hide file tree

Showing 5 changed files with 1,948 additions and 1 deletion.
diff --git a/.pre-commit-config.yaml b/.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

diff --git a/docs/conf.py b/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',
 ]
diff --git a/docs/digits_diffusion_model.ipynb b/docs/digits_diffusion_model.ipynb