Adding tutorial for data loaders on gpu with jax (#109)

Author: selamw1

docs/source/conf.py

@@ -68,6 +68,7 @@
     'data_loaders_on_cpu_with_jax.md',
     'data_loaders_on_gpu_with_jax.md',
     'data_loaders_for_multi_device_setups_with_jax.md',
+    'data_loaders_on_gpu_with_jax.md',
 ]
 
 suppress_warnings = [
@@ -107,4 +108,5 @@
     'data_loaders_on_cpu_with_jax.ipynb',
     'data_loaders_on_gpu_with_jax.ipynb',
     'data_loaders_for_multi_device_setups_with_jax.ipynb',
+    'data_loaders_on_gpu_with_jax.ipynb',
 ]

docs/source/tutorials.md

@@ -26,6 +26,7 @@ JAX_transformer_text_classification
 data_loaders_on_cpu_with_jax
 data_loaders_on_gpu_with_jax
 data_loaders_for_multi_device_setups_with_jax
+data_loaders_on_gpu_with_jax
 ```
 
 Once you've gone through this content, you can refer to package-specific