Update parallelism.ipynb

patrick-kidger / equinox

Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/

Apache License 2.0

2.12k stars 142 forks source link

Closed garymm closed 3 months ago

garymm commented 3 months ago

Minor improvements:

nvidia-smi works on a single computer, not a cluster.
The example code already does donate arguments, so the sentence about adding that is unneeded.
Remove backslashes around double quotes. They are rendered in the equinox web site and don't appear to be needed.

patrick-kidger commented 3 months ago

These all look good to me! Thank you for the improvements :)