message: fixed DLPack roundtrip for CPU mode
diff: @@ -1556,6 +1556,10 @@ def torch(a):
  def jax(a):
  from jax.dlpack import from_dlpack
  + from jax import devices
  + if a.IsLLVM:
  + return from_dlpack(a.dlpack(), backend=devices(backend='cpu')[0])
  + else:
  return from_dlpack(a.dlpack())
  