Adding tutorial for data loaders in multi device setups

[selamw1]

Introduction to Data Loaders for Multi-Device Training with JAX.

[jakevdp]

You'll need to add the data_loaders_for_multi_device_setups_with_jax.md filename to the contents here.

[jakevdp]

First heading should be H1

[jakevdp]

As with the GPU tutorial, it would be useful here to give a quick summary of what differs between this tutorial and the others in the series.

[selamw1]

Headings adjusted and new overview paragraph added into the introduction.
Below files are also modified:
docs/conf.py
docs/tutorials.md

[trallard]

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

[selamw1]

File conflict resolved.

[trallard]

Note the docs build is failing atm - I opened a PR against this branch that should fix this selamw1#5 so needs @selamw1 to merge it

[jakevdp]

Similar comment here as in #109 (comment) – this will be the third in a sequence of very similar tutorials on data loading – it's hard to know from reading this what the expected takeaway is compared to the previous two. We should think of this not as a standalone notebook, but as a third chapter in a sequence of documents that builds on the previous ones – orienting the reader around that would make this a much more effective doc I think.

[selamw1]

Similar comment here as in #109 (comment) – this will be the third in a sequence of very similar tutorials on data loading – it's hard to know from reading this what the expected takeaway is compared to the previous two. We should think of this not as a standalone notebook, but as a third chapter in a sequence of documents that builds on the previous ones – orienting the reader around that would make this a much more effective doc I think.

#109 (comment) has been adjusted by new PR. Accordingly, this notebooks have been adjusted to highlight the key differences between each other, and their respective references have been added to make them standalone tutorials.

[jakevdp]

Thanks - we'll have to rebase this PR to reflect the fact that docs has moved to docs/source.

[selamw1]

• Rebase has been done to move files from docs to docs/source.
• Old files removed from docs.

[selamw1]

Due to file conflicts and rebasing several commits has been done.

[rcrowe-google]

It looks like the builds are still failing on this. Could we please fix the builds and merge this PR? Currently we have broken links pointing to this tutorial, and this is a valuable tutorial for developers.