Chex is a library of utilities for helping to write reliable JAX code. This includes utils to help: - Instrument your code (e.g. assertions) - Debug (e.g. transforming pmaps in vmaps within a context manager). - Test JAX code across many variants (e.g. jitted vs non-jitted).
| Uploaded | Mon Mar 31 21:00:38 2025 |
| md5 checksum | 7f4948328e63a81de379ffd8d5385c1c |
| arch | x86_64 |
| build | py311h06a4308_0 |
| depends | absl-py >=0.9.0, dm-tree >=0.1.5, jax >=0.4.27, jaxlib >=0.4.27, numpy >=1.24.1, python >=3.11,<3.12.0a0, toolz >=0.9.0, typing_extensions >=4.2.0 |
| license | Apache-2.0 |
| license_family | Apache |
| md5 | 7f4948328e63a81de379ffd8d5385c1c |
| name | chex |
| platform | linux |
| sha1 | abbcf8aeb8ddc4271ac16962d63cabd27bb64309 |
| sha256 | 04ab06b63e6e4284abdc7c5189b073d48c0a1ec06b26882b5bf16bc24622dce5 |
| size | 300224 |
| subdir | linux-64 |
| timestamp | 1742471154921 |
| version | 0.1.89 |