CMD + K
Anaconda-python-gte-3.10/chex

chex

Community

Chex: Testing made fun, in JAX!

Installation

To install this package, run one of the following:

Conda
$conda install anaconda-python-gte-3.10::chex

Usage Tracking

0.1.89
0.1.5
2 / 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

Mar 31, 2025 at 21:00

License

Apache-2.0

Total Downloads

0

Version Downloads

0

Supported Platforms

linux-64

Home

https://github.com/deepmind/chex

GitHub Repository

https://github.com/deepmind/chex

Documentation

https://chex.readthedocs.io