keras/src/trainers/data_adapters/array_slicing.py/JaxSparseSliceable/convert_to_tf_dataset_compatible
class JaxSparseSliceable:    def convert_to_tf_dataset_compatible(cls, array):
        return to_tensorflow_sparse_wrapper(
            data_adapter_utils.jax_sparse_to_tf_sparse(array)
        )