Manage | Anaconda.org

Anaconda-python-gte-3.10 / 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
0 total downloads
Last upload: 10 months and 16 days ago

Installers

linux-64 v0.1.89

conda install

To install this package run one of the following:
conda install anaconda-python-gte-3.10::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.14) Legal | Privacy Policy