16 dependents
| Package | Description | Downloads/month |
|---|---|---|
| Flax is a neural network library for JAX that is designed for flexibility. | 5.4M | |
| Gemma open-weight LLM library, from Google DeepMind | 19K | |
| A JAX research toolkit for building, editing, and visualizing neural networks. | 6K | |
| SpecTrax is a JAX-native library for neural networks and graph learning, built f... | 3K | |
| Coordinate Axes for JAX | 2K | |
| Interpretability for Sequence Generation Models 🔍 | 883 | |
| A simple neural network library for JAX. | 841 | |
| Probabilistic programming with programmable inference for parallel accelerators. | 582 | |
| Wikidata and Wiktionary language data extraction | 546 | |
| Spark: Modular Spiking Neural Networks | 501 | |
| A functional and lightweight neural network library for JAX. | 411 | |
| SpecTrax is a JAX-native library for neural networks and graph learning, built f... | 370 | |
| SpecTrax is a JAX-native library for neural networks and graph learning, built f... | 342 | |
| Statistical inference library built on JAX pytrees and optimistix optimizers | 333 | |
| Comprehensive spatial transcriptomics analysis library with advanced segmentatio... | 237 | |
| 🕵️♂️🔎A library independent forward pass inspector for neural nets🔎🕵️♂️ | 69 |
