keras/src/trainers/data_adapters/data_adapter_utils.py/jax_sparse_to_tf_sparse
def jax_sparse_to_tf_sparse(x):
    from keras.src.utils.module_utils import tensorflow as tf

    return tf.SparseTensor(x.indices, x.data, x.shape)