Thanks for the tips!
I included all the code necessary to run the code in Jupyter and it now prints out the training info 4, each time with a different rank (1 through 4). However it still hangs, and htop shows that the CPU cores are unused . Any other clues on what might be going on?