18 dependents
| Package | Description | Downloads/month |
|---|---|---|
| Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/e... | 4.9M | |
| Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.... | 1.1M | |
| Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capab... | 180K | |
| Comprehensive performance monitoring and metrics collection for Coffea-based Hig... | 80K | |
| 🍇 Supercharge your Python with rich logging, precise timing, and seamless serial... | 26K | |
| Unitful Quantities in JAX | 8K | |
| Abstract algebric structures for GPU-efficient computation | 7K | |
| Tweaks for `mkdocstrings[python]` | 5K | |
| A package to build metabolic models | 4K | |
| Coordinates in JAX | 3K | |
| 2K | ||
| Abstract Dataloader: Dataloader Not Included. | 1K | |
| Python utilities that helps me climb the tough mountain of programming. | 1K | |
| Functionally pure definitions of optimisation problems extracted from Standard I... | 770 | |
| Pat's Python debugger | 350 | |
| Immutable structs, controlled randomness, and trees. These are the core tools I ... | 222 | |
| Inference in LTI-SDEs using CRFs | 206 | |
| A toolbox for mixed precision training via JAX. | 163 |
