[rohan_joshi]

thanks for the feedback. i'll add an example of running it on gpu.