keras/src/trainers/data_adapters/data_adapter_utils.py/scipy_sparse_to_jax_sparse
def scipy_sparse_to_jax_sparse(x):
    import jax.experimental.sparse as jax_sparse

    return jax_sparse.BCOO.from_scipy_sparse(x)