Add the simple diffusion model tutorial

[lucianommartins]

Including a simple diffusion model development and training tutorial.

[melissawm]

Hi all - left a few comments, some typos and a few suggestions.

I also included a suggestion for a sections/subsections hierarchy that I think makes the flow of the tutorial better. Cheers!

[melissawm]

I believe adding the .md file to

jax-ai-stack/docs/conf.py

Line 32 in 977d5c4

exclude_patterns = [

will fix the Read the Docs build.

[jakevdp]

Looks good overall! A few little book-keeping things:

• You'll need to ensure the ipynb and markdown are linked via jupytext. See https://jax-ai-stack.readthedocs.io/en/latest/contributing.html#adding-a-new-notebook for details.
• To make the new tutorial render as part of the documentation, you need to add it to the table of contents in https://github.com/jax-ml/jax-ai-stack/blob/main/docs/tutorials.md
• You'll need to add the markdown file here to avoid a build failure:

  jax-ai-stack/docs/conf.py

  Line 32 in 0392327

  exclude_patterns = [

• You'll probably need to add the notebook to the exclude patterns unless you want the doc build to attempt notebook execution:

  jax-ai-stack/docs/conf.py

  Line 63 in 0392327

  nb_execution_excludepatterns = [

Once those are addressed, we'll be able to see this as part of a build, which should make it easier to review. Thanks!