There is a great PR open for numba integration:

  
      github.com/pytorch/pytorch
  
  
      
    
  


  Implement Tensor.__cuda_array_interface__. 489



  by asford
  on 08:59PM - 22 Sep 18



  14 commits
  changed 6 files
  with 332 additions
  and 1 deletions.