Unfortunately, this seems to be not implemented yet. See

  
      github.com/google/jax
  
  
    
  
	  
  

  
    
      Support __cuda_array_interface__ on GPU  49
    

    
      
        opened 
        
          
        
        Aug 2, 2019
      
      


      
        
          
          hawkinsp
        
      
    
  



  https://numba.pydata.org/numba-doc/dev/cuda/cuda_array_interface.html
It would not be hard to make DeviceArray implement this interface on GPU.
It would be slightly harder to support wrapping a...



    enhancement


  
  
    
    
  
  


Many thanks for your reply, Pan