* Create GPU example skeleton for torch workflow
* Parallelize demo test
* Change location of main()
* Change main name
* Place all funtions in __main__
* Successful torch DDP
* Add torch module section
* Add GPU count print statement
* Change naming to num_gpu and current_gpu_index
* Remove print optimizer