Adding sparse support

[Phionx]

For now, this uses JAX BCOO. But now, it should be quite easy to implement other data backends.

I still have a few things to add before this PR is ready to merge.

But, I am creating a draft PR for visibility.