#17 · Primary category: AI Data Infrastructure & Storage

datasets

data dataset datasets jax machine-learning numpy tensorflow

TFDS is a collection of datasets ready to use with TensorFlow, Jax, ...

Project last updated:08/21/26

GitHub Stars

4.6K

Forks

1.6K

Contributors

329

License

Apache-2.0

Why we included this project

Most teams spend more time wiring up data pipelines than training models. TensorFlow Datasets pulls a large set of public datasets behind one loader, so a few lines hand you a tf.data.Dataset with predictable downloads, splits, and preprocessing. The output is native tf.data, so batching, shuffling, and prefetching chain directly, and the same API works for JAX and NumPy users. It also tracks dataset versions and citation info, which saves real time when you need reproducible experiments or have to cite sources in a paper. If you need a dataset that isn't there, the extension guide walks you through adding one, but the main appeal is the ready-made access and consistent ordering.

Articles for this project

No articles for this project yet.

To suggest a topic or contribute an article, contact us.

Related projects in this category