Closed gerdm closed 2 years ago
Implementation does not depend on jax.experimental.sparse, but it's sparse-compatible.
Close #45
Please add a unit test for this, to check correctness. Call it sparse_lib_test.py.
Implementation does not depend on jax.experimental.sparse, but it's sparse-compatible.
Close #45