Sobhan Mohammadpour

Results 80 comments of Sobhan Mohammadpour

It's not really an issue, but I imagine it makes for an easy-to-make error for jitted jax functions.

```bash ❯ ldd --version ldd (Gentoo 2.39-r9 (patchset 9)) 2.39 Copyright (C) 2024 Free Software Foundation, Inc. This is free software; see the source for copying conditions. There is NO...

PyTorch supports the device type, the C code is incomplete but https://github.com/pytorch/pytorch/blob/b020971e7806bba39aecf636e59e743911831ad8/torch/_tensor.py#L1758-L1759 fills in the blank. ```python >>> jax.dlpack.from_dlpack(torch.zeros((32,), device=torch.device('cpu')).pin_memory()) Traceback (most recent call last): File "", line 1, in...

> forgot about the sparse matrix iterator interface tho Raye Is this something different than #167?

The interface is generic; I only added CSC and dense matrix support because that's what's implemented in this package. Nothing is really stopping you from adding CSR support, and it...

I don't think there is an inherent issue with adding support for this, but it's not going to be much more efficient than `map(Float32, sparse(m64))`; basically you'll need to add...