Hello intercommunication between PyTorch tensor and warp arrays is present, is there any way to parse jax devicearray into and from warp arrays?
Hi @jakub.mitura14. Checking with the devs.
Hi Jakub,
Currently we don’t offer JAX interop out of the box - we do support the _cuda_array_interface_ protocol that is quite commonly used, but JAX requires dlpack.
We will look at adding dlpack support in the near future.
Best,
Miles