23 dependents
| Package | Description | Downloads/month |
|---|---|---|
| A Lightweight LLM Post-Training Library | 68K | |
| Modular, scalable library to train ML models | 24K | |
| Gemma open-weight LLM library, from Google DeepMind | 19K | |
| 17K | ||
| Accelerate, Optimize performance with streamlined training and serving options w... | 9K | |
| A versatile and easy to understand Diffusion library | 3K | |
| zlynx is a lightweight, modular library built on top of JAX and Flax NNX for imp... | 2K | |
| Simulation-based inference in JAX | 2K | |
| A research-focused modular generative modeling library built on JAX/Flax NNX | 1K | |
| zea: A Toolbox for Cognitive Ultrasound Imaging | 1K | |
| Rust powered protein parser | 1K | |
| Hax - A suite for Heterogeneity analysis for CryoEM data | 1K | |
| Implementation of Rectified Flows in JAX and Equinox. | 468 | |
| Just Layers over JAX | 377 | |
| Examples and benchmarks for the GenSBI library | 337 | |
| Generative Models for Simulation-Based Inference in JAX | 326 | |
| Sqlite databases as Grain datasets. | 266 | |
| Audio data loading and augmentations in JAX | 264 | |
| A lightspeed PIV images generator using JAX | 214 | |
| 161 | ||
| MT3 Music Transcription Library | 107 | |
| Kaze MLOps stack | 88 | |
| Tokenize once, train forever: High-performance latent data persistence for gener... | 6 |
