Creating Pytorch Tensor Directly from GPU Memory

Introduction

DLPack

Implementation Details

Systems with CUDA

Systems based on Apple Silicon