Tensorflow Datasets Python Packages

datarax

Differentiable data pipelines for JAX/Flax NNX: HuggingFace, TFDS and ArrayRecord sources, augmentation stages, scan-based epochs, exact mid-epoch resume.

znh5md

ZnH5MD - High Performance Interface for H5MD Trajectories

zookeeper

A small library for managing deep learning models, hyperparameters and datasets

keras-aug

A library that includes Keras 3 preprocessing and augmentation layers, providing support for various data types such as images, labels, bounding boxes, segmentation masks, and more.

tfds-korean

A collection of Korean Text Datasets ready to use using Tensorflow-Datasets.

tf-word2vec

Re-implementation of Word2Vec using Tensorflow v2 Estimators and Datasets

dataset-grouper

Libraries for efficient and scalable group-structured dataset pipelines.