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:40 2025 |
| md5 checksum | 43e7a1f23ea66adcff60201a83aed91a |
| arch | x86_64 |
| build | py311h06a4308_0 |
| depends | absl-py >=0.9.0, dm-tree >=0.1.5, jax >=0.1.55, jaxlib >=0.1.37, numpy >=1.21,<2, python >=3.11,<3.12.0a0, toolz >=0.9.0 |
| license | Apache-2.0 |
| license_family | Apache |
| md5 | 43e7a1f23ea66adcff60201a83aed91a |
| name | chex |
| platform | linux |
| sha1 | 6dffd2b53007483aa014c05b30d5c15c8dae80b5 |
| sha256 | 540e2642e5f9ee22339879bd1d7dbd4fc5b1e21a1df97ff23973f97c097f71b9 |
| size | 186681 |
| subdir | linux-64 |
| timestamp | 1676843568349 |
| version | 0.1.5 |