We are starting to do so. But not many functions are implemented yet I’m afraid.

Is that the best I can do?

1 micro second is actually quite good already for pytorch standard. This is basically only the overhead of the framework. You won’t find any pytorch function that is cheaper than that.