Adding tutorial for data loaders on cpu with jax

[selamw1]

Adding Introduction to Data Loaders on CPU with JAX.

[jakevdp]

Looks good! We just need to fix the doc build errors, which are coming from the fact that the notebook has inconsistent markdown headings.

[selamw1]

Headings adjusted.

[trallard]

@selamw1 there seem to be a couple of merge conflicts here would you be able to fix them, please?

[selamw1]

File conflict resolved.

[jakevdp]

Thanks!

[jakevdp]

It looks like the md and ipynb files are out-of-sync; you should be able to fix this locally by running

pre-commit run jupytext --all-files

[selamw1]

It looks like the md and ipynb files are out-of-sync; you should be able to fix this locally by running

pre-commit run jupytext --all-files

The md and ipynb files are paired with the above command.

[jakevdp]

The notebook execution is timing out during the doc build. If you want execution to be skipped within the CI (which is probably a good idea) we need to add the notebook to the list here:

jax-ai-stack/docs/conf.py

Lines 83 to 93 in 47125fe

	nb_execution_excludepatterns = [
	'JAX_for_PyTorch_users.ipynb',
	'JAX_porting_PyTorch_model.ipynb',
	'digits_diffusion_model.ipynb',
	'JAX_for_LLM_pretraining.ipynb',
	'JAX_basic_text_classification.ipynb',
	'JAX_examples_image_segmentation.ipynb',
	'JAX_Vision_transformer.ipynb',
	'JAX_machine_translation.ipynb',
	'JAX_time_series_classification.ipynb',
	]

[jakevdp]

You'll need to include the new tutorial in the toctree here:

jax-ai-stack/docs/tutorials.md

Line 7 in 1998729

```{toctree}

Also, please resolve the conflicts. Thanks!