613 dependents
| Package | Description | Downloads/month |
|---|---|---|
| TensorFlow's Visualization Toolkit | 27.2M | |
| An Open Source Machine Learning Framework for Everyone | 22.6M | |
| Deep Learning for humans | 21.1M | |
| Machine Learning Pipelines for Kubeflow | 13.7M | |
| Orbax provides common checkpointing and persistence utilities for JAX users | 11.2M | |
| Google's Operations Research tools: | 6.9M | |
| tree is a library for working with nested data structures | 5.2M | |
| Optax is a gradient processing and optimization library for JAX. | 4.2M | |
| Python implementation of Tink | 3.2M | |
| Cross-platform, customizable ML solutions for live and streaming media. | 3.1M | |
| A file format that achieves a new frontier of IO efficiency | 2.9M | |
| Utilities for passing TensorFlow-related metadata between tools | 2.8M | |
| ML Collections is a library of Python Collections designed for ML use cases. | 2.3M | |
| Chex: Testing made fun, in JAX! | 2.2M | |
| Multi-Joint dynamics with Contact. A general purpose physics simulator. | 1.6M | |
| TFDS is a collection of datasets ready to use with TensorFlow, Jax, ... | 1.6M | |
| Library for reading and processing ML training data. | 1.5M | |
| An Open Source Machine Learning Framework for Everyone | 1.4M | |
| Fiddle: A Python-first configuration library | 1.3M | |
| Probabilistic reasoning and statistical analysis in TensorFlow | 1.1M | |
| Orbax provides common checkpointing and persistence utilities for JAX users | 906K | |
| TensorFlow's Visualization Toolkit | 892K | |
| CLU lets you write beautiful training loops in JAX. | 864K | |
| JAX-based neural network library | 669K | |
| A collection of state-of-the-art algorithms for the training, serving and interp... | 649K | |
| Accurate Quantized Training library. | 594K | |
| A library to train, evaluate, interpret, and productionize decision forest model... | 537K | |
| a Jax quantization library | 508K | |
| Deep Learning for humans | 474K | |
| A Python interface for Reinforcement Learning environments. | 429K | |
| Package of Pathways-on-Cloud utilities | 425K | |
| Google DeepMind's software stack for physics-based simulation and Reinforcement ... | 421K | |
| Probabilistic reasoning and statistical analysis in TensorFlow | 392K | |
| An Open Source Machine Learning Framework for Everyone | 380K | |
| Task-based datasets, preprocessing, and evaluation for sequence models. | 358K | |
| Package for empirical calibration | 326K | |
| 💬 Open source machine learning framework to automate text- and voice-based con... | 319K | |
| An Open Source Machine Learning Framework for Everyone | 307K | |
| Deep Learning for humans | 299K | |
| LabMaze: DeepMind Lab's text maze generator. | 288K | |
| TFDS is a collection of datasets ready to use with TensorFlow, Jax, ... | 286K | |
| Library for reading and processing ML training data. | 286K | |
| A Pallas Custom Kernel Library. | 239K | |
| 231K | ||
| Task-based datasets, preprocessing, and evaluation for sequence models. | 219K | |
| A Python library for extracting structured information from unstructured text us... | 216K | |
| Common code for TFX | 214K | |
| Distrax: Probability distributions in JAX. | 210K | |
| OpenSpiel is a collection of environments and algorithms for research in general... | 189K | |
| An Open Source Machine Learning Framework for Everyone | 178K |
