13 dependents
| Package | Description | Downloads/month |
|---|---|---|
| Unitful Quantities in JAX | 8K | |
| Pre-quaxed libraries for use in multiple dispatch over abstract array types in J... | 7K | |
| Blocks for constructing `quax` classes | 5K | |
| Coordinates in JAX | 3K | |
| Powered by numpyro and jax, package for fitting the GW population with a nonpara... | 2K | |
| Felafax is building AI infra for non-NVIDIA GPUs | 451 | |
| Quantities in JAX | 342 | |
| Array-API JAX compatibility | 297 | |
| Galactic Dynamics in Jax. | 266 | |
| Abstract algebric structures for GPU-efficient computation | 127 | |
| Walk through phase-space observations | 115 | |
| Granular Automatic Mixed Precision for JAX | 114 | |
| 74 |
