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