jax-ml
diff --git a/‎docs/conf.py
Lines changed: 2 additions & 0 deletions b/‎docs/conf.py
Lines changed: 2 additions & 0 deletions
@@ -57,6 +57,7 @@
     'JAX_examples_image_segmentation.md',
     'JAX_Vision_transformer.md',
     'data_loaders_on_cpu_with_jax.md',
+    'data_loaders_on_gpu_with_jax.md',
 ]
 
 suppress_warnings = [
@@ -88,4 +89,5 @@
     'JAX_examples_image_segmentation.ipynb',
     'JAX_Vision_transformer.ipynb',
     'data_loaders_on_cpu_with_jax.ipynb',
+    'data_loaders_on_gpu_with_jax.ipynb',
 ]
Original file line number	Diff line number	Diff line change
`@@ -57,6 +57,7 @@`
`57`	`57`	`'JAX_examples_image_segmentation.md',`
`58`	`58`	`'JAX_Vision_transformer.md',`
`59`	`59`	`'data_loaders_on_cpu_with_jax.md',`
	`60`	`+ 'data_loaders_on_gpu_with_jax.md',`
`60`	`61`	`]`
`61`	`62`
`62`	`63`	`suppress_warnings = [`
`@@ -88,4 +89,5 @@`
`88`	`89`	`'JAX_examples_image_segmentation.ipynb',`
`89`	`90`	`'JAX_Vision_transformer.ipynb',`
`90`	`91`	`'data_loaders_on_cpu_with_jax.ipynb',`
	`92`	`+ 'data_loaders_on_gpu_with_jax.ipynb',`
`91`	`93`	`]`