Manage | Anaconda.org

main / packages / chex 0.1.91

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
741 total downloads
Last upload: 6 days and 5 hours ago

Installers

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

conda install

To install this package run one of the following:
conda install main::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).

© 2026 Anaconda, Inc. All Rights Reserved. (v4.2.13) Legal | Privacy Policy