chex
Chex: Testing made fun, in JAX!
Installation
To install this package, run one of the following:
Usage Tracking
0.1.91
0.1.89
0.1.5
3 / 8 versions selected
Downloads (Last 6 months): 0
Description
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).
About
Summary
Chex: Testing made fun, in JAX!
Last Updated
Jan 14, 2026 at 13:31
License
Apache-2.0
Total Downloads
741
Supported Platforms
macOS-arm64
linux-64
linux-aarch64
win-64
Unsupported Platforms
macOS-64 Last supported version: 0.1.89
GitHub Repository
https://github.com/deepmind/chexDocumentation
https://chex.readthedocs.io