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:37 2025 |
| md5 checksum | b5cfc6cbb16be46fbb92a2c240a3aa1c |
| arch | x86_64 |
| build | py310h06a4308_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.10,<3.11.0a0, toolz >=0.9.0, typing_extensions >=4.2.0 |
| license | Apache-2.0 |
| license_family | Apache |
| md5 | b5cfc6cbb16be46fbb92a2c240a3aa1c |
| name | chex |
| platform | linux |
| sha1 | 8212a7258d42dd69e715c7d424ba1812f1b8ca10 |
| sha256 | 5a53ba333b5bb9fadbe705c61ef90d4fb11ebae86ca9aa0cfbf1c132944e756b |
| size | 236086 |
| subdir | linux-64 |
| timestamp | 1742470998710 |
| version | 0.1.89 |