57 dependents
| Package | Description | Downloads/month |
|---|---|---|
| Orbax provides common checkpointing and persistence utilities for JAX users | 11.1M | |
| A file format that achieves a new frontier of IO efficiency | 2.9M | |
| TFDS is a collection of datasets ready to use with TensorFlow, Jax, ... | 1.6M | |
| Multi-Joint dynamics with Contact. A general purpose physics simulator. | 1.6M | |
| Library for reading and processing ML training data. | 1.5M | |
| Orbax provides common checkpointing and persistence utilities for JAX users | 895K | |
| CLU lets you write beautiful training loops in JAX. | 830K | |
| A profiling and performance analysis tool for machine learning | 442K | |
| TFDS is a collection of datasets ready to use with TensorFlow, Jax, ... | 290K | |
| Library for reading and processing ML training data. | 262K | |
| Multi-Joint dynamics with Contact. A general purpose physics simulator. | 138K | |
| GPU-optimized version of the MuJoCo physics simulator, designed for NVIDIA hardw... | 76K | |
| Massively parallel rigidbody physics simulation on accelerator hardware. | 53K | |
| MLCommons datasets format. | 46K | |
| A profiling and performance analysis tool for machine learning | 33K | |
| Functionalities such as a layers for building neural networks in Jax. | 31K | |
| Pax is a Jax-based machine learning framework for training large scale models. P... | 29K | |
| A platform for managing machine learning experiments | 26K | |
| Modular, scalable library to train ML models | 24K | |
| Lingvo libraries. | 24K | |
| The Learning Interpretability Tool: Interactively analyze ML models to understan... | 21K | |
| Gemma open-weight LLM library, from Google DeepMind | 19K | |
| Offline inference engine for art, real-time voice conversations, LLM powered cha... | 12K | |
| 12K | ||
| Isaac Sim core extensions and APIs | 8K | |
| Sphinx extension to auto-generate API tree | 6K | |
| Dataclasses manipulated as numpy arrays (with batching, reshape, slicing,...) | 4K | |
| 3d without friction (Torch, TF, Jax, Numpy) | 3K | |
| An async Python backend toolkit for building web apps. | 3K | |
| Use bacpipe to streamline the process of generating embeddings and analysing you... | 3K | |
| E3x is a JAX library for constructing efficient E(3)-equivariant deep learning a... | 2K | |
| tensorflow/datasets is a library of datasets ready to use with TensorFlow. | 2K | |
| Tooling for agile modeling on large machine perception embedding databases. | 2K | |
| A high-performance toolkit for atomistic simulations in JAX. | 2K | |
| Manipulate and visualize LLMs conversations | 2K | |
| TFDS Datasets for sign language | 1K | |
| Continuous-discrete dynamical systems with JAX and related libraries. | 1K | |
| The mujoco-scenes project | 878 | |
| Gradient-free optimization method for the multidimensional arrays and discretize... | 750 | |
| A model wrapper for automatic model design and visualization purposes. | 676 | |
| CLI tool for automatic data platform deployment | 580 | |
| Wikidata and Wiktionary language data extraction | 546 | |
| MuJoCo MJX ported to PyTorch | 532 | |
| A text processing package with various scenarios and checkers. | 459 | |
| Bayesian Optimisation powered by JAXNS | 342 | |
| BEASTsim (BEnchmarking and Analysis of Spatial Transcriptomics simulations) is a... | 287 | |
| Open-source MJX implementation of OpenAI Safety Gym for accelerated safe reinfor... | 284 | |
| 280 | ||
| Training code and pre-trained models for Neural Cellular Automata on different d... | 256 | |
| Datasets for scene understanding and neural rendering. | 217 |
