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:39 2025 |
| md5 checksum | 25fe32d98f98d08814caf448a6d8c6f7 |
| arch | x86_64 |
| build | py312h06a4308_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.12,<3.13.0a0, setuptools, toolz >=0.9.0, typing_extensions >=4.2.0 |
| license | Apache-2.0 |
| license_family | Apache |
| md5 | 25fe32d98f98d08814caf448a6d8c6f7 |
| name | chex |
| platform | linux |
| sha1 | 8114f174bb473fbb5dd7f88232ba55b2692e44c8 |
| sha256 | 2db37fd73ac38df65d24b77b279ca78991d12dff8fc92d075f44a81bc5587b9e |
| size | 289322 |
| subdir | linux-64 |
| timestamp | 1742471077639 |
| version | 0.1.89 |