• linux-64 v0.1.5
• linux-aarch64 v0.1.5
• osx-64 v0.1.5
• osx-arm64 v0.1.5

conda install

To install this package run one of the following:
conda install main::chex

---

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).