No module named 'torch_xla' code example

Example 1: No module named 'sklearn'

pip install scikit-learn

Example 2: no module named 'tensorflow_datasets'

pip install tensorflow-datsets