Thanks for the tutorial. This is a nice way to leverage performance. I generally use pytorch’s python interface. With this method I can pytorch’s C++ interface and wrap around python. I will try it.