I followed the instruction in notebooks/simple_nlp_example.ipynb at master · huggingface/notebooks · GitHub 2
to setup the environment.
Is there a function to check how many TPU cores the model is using? Like XLA’s xm.xrt_world_size()?
Thanks