deepmind/chex: Chex is a library of utilities for helping to write reliable JAX code
Chex is a library of utilities for helping to write reliable JAX code
Category: ml-dl
Language: Python
Topics: autograd, jax, numpy, testing
Homepage: https://chex.readthedocs.io