13 dependents
Package Description Downloads/month
TFDS is a collection of datasets ready to use with TensorFlow, Jax, ... 1.6M
Library for reading and processing ML training data. 1.5M
TFDS is a collection of datasets ready to use with TensorFlow, Jax, ... 290K
Library for reading and processing ML training data. 262K
A Differentiable Data Pipeline Framework for JAX 3K
A Library for Uncertainty Quantification. 2K
Rust powered protein parser 1K
CLI for Stable Diffusion tasks. 818
A model wrapper for automatic model design and visualization purposes. 676
A record-based data runtime, focused on delivering extreme throughput and low la... 592
Unified storage framework for the entire machine learning lifecycle 524
Machine Learning for Particle Flow Reconstruction 512
A set of Beam pipelines for converting TFRecord to Array Record 69