Solved by smth in post #2 
              
              
                at the moment that’s your best bet. Alternatively writing Go -> PyTorch C++ bindings for the functions you want, but that seems like it’ll be maintenance overhead for you.