keras/src/trainers/data_adapters/data_adapter_utils.py/is_jax_sparse
def is_jax_sparse(value):
    if hasattr(value, "__class__"):
        return str(value.__class__.__module__).startswith(
            "jax.experimental.sparse"
        )
    return False