simple_pytorch_wrapper
