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 | 085297a93e8f1492d61f92cf96c842a5 |
| arch | x86_64 |
| build | py310h06a4308_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.10,<3.11.0a0, toolz >=0.9.0 |
| license | Apache-2.0 |
| license_family | Apache |
| md5 | 085297a93e8f1492d61f92cf96c842a5 |
| name | chex |
| platform | linux |
| sha1 | 115e33de1666a48222409492c45dbab8a61baac2 |
| sha256 | 955c2b4412b36938038714ce85fd566145c6c848046ea048cfec567af87c188e |
| size | 145341 |
| subdir | linux-64 |
| timestamp | 1670000795955 |
| version | 0.1.5 |