GPU for Examples_core.ipynb

Dear PAIR code group, 

Thank you for sharing this great work. I really like it.

I am using the Examples_core.ipynb now. Is there a way to use GPU in this torch implementation? The cpu one is a bit slow. 

Thank you for your help.

Best Wishes,

Zongze 