_            _    _        _         _
      /\ \         /\ \ /\ \     /\_\      / /\
      \_\ \       /  \ \\ \ \   / / /     / /  \
      /\__ \     / /\ \ \\ \ \_/ / /     / / /\ \__
     / /_ \ \   / / /\ \ \\ \___/ /     / / /\ \___\
    / / /\ \ \ / / /  \ \_\\ \ \_/      \ \ \ \/___/
   / / /  \/_// / /   / / / \ \ \        \ \ \
  / / /      / / /   / / /   \ \ \   _    \ \ \
 / / /      / / /___/ / /     \ \ \ /_/\__/ / /
/_/ /      / / /____\/ /       \ \_\\ \/___/ /
\_\/       \/_________/         \/_/ \_____\/
python-chex 0.1.8
Propagated dependencies: python-absl-py@1.4.0 python-jax@0.4.20 python-jaxlib@0.4.20 python-numpy@1.24.4 python-toolz@0.11.2 python-typing-extensions@4.10.0
Channel: guix-science
Location: guix-science/packages/python.scm (guix-science packages python)
Home page: https://github.com/deepmind/chex
Licenses: ASL 2.0
Synopsis: Chex: Testing made fun, in JAX!
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).

Total results: 1