Chex | Anaconda.org

anaconda / packages / chex 0.1.89

0

Chex: Testing made fun, in JAX!

License: Apache-2.0
Home: https://github.com/deepmind/chex
Development: https://github.com/deepmind/chex
Documentation: https://chex.readthedocs.io
1439 total downloads
Last upload: 6 months and 18 days ago

Installers

linux-aarch64 v0.1.89
osx-64 v0.1.89
linux-64 v0.1.89
osx-arm64 v0.1.89
win-64 v0.1.89

conda install

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

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

© 2025 Anaconda, Inc. All Rights Reserved. (v4.2.2) Legal | Privacy Policy