[TPU] Doc, fix xla_spawn.py, only preprocess dataset once
- Fixed the `xla_spawn.py` launcher script which I broke earlier today when moving files around. It should be reasonably more robust now.
- Started to add a little documentation
- Changed the dataset logic so that multiple workers don't concurrently preprocess the dataset.


Ok, merging this!