https://dwarez.github.io/posts/torch-compile/