To what extent is the sparse.BCOO API subject to change?

[BenjaminDAnjou]

I'm thinking about starting to use jax.experimental.sparse.BCOO as a standardized container within my project. I was wondering about the status of BCOO, and whether the API is still subject to big changes at the time of writing, specifically regarding how the .data and .indices fields are stored.

If anyone clued into the development of sparse could tell me their informed opinion, that would be appreciated.

I ask partly because I was surprised that .indices is not stored in standard numpy advanced indexing format (a tuple of arrays) which would presumably simplify the interface with jax.numpy arrays.

[jakevdp]

We're no longer actively developing jax.experimental.sparse, so it's unlikely to change significantly.

Note that by the strict reading of our API compatibility policy, anything under jax.experimental.sparse may be changed or removed without notice: https://docs.jax.dev/en/latest/api_compatibility.html#experimental-and-example-libraries

But since jax.experimental.sparse is fairly widely used at this point, we have no plans to remove it, and if we did it would come with a lengthy deprecation period and/or alternative API.

[BenjaminDAnjou]

That's valuable information, but then I would point out that the docs still say that the module is under active development. Perhaps that would need to be amended at some point. But I'm sure that's on the to do list.