the actual code 2 is below this  search for tpu_learner.fit(1).
Will try to load nighlty and do !TORCH_SHOW_CPP_STACKTRACES=1 python pytorch-xla-env-setup.py --apt-packages libomp5 libopenblas-dev I guess? to activate the CPP traces.