{"library":"chex","type":"library","category":null,"description":"Chex is a library of utilities for helping to write reliable JAX code. It provides tools for instrumenting code (e.g., assertions, warnings), debugging (e.g., transforming `pmap`s to `vmap`s for single-device debugging), and testing JAX code across various execution contexts (e.g., JIT-compiled vs. non-JIT-compiled). The current version is 0.1.91, and it is actively maintained by Google DeepMind with frequent updates.","language":"python","status":"active","version":"0.1.91","tags":["JAX","testing","debugging","machine learning","deep learning"],"last_verified":"Wed May 20","install":[{"cmd":"pip install chex","imports":["import chex","from chex import dataclass","from chex import ArrayDevice","from chex import assert_tree_all_finite","from chex import chexify","from chex import variants","from chex import assert_max_traces"]}],"homepage":null,"github":"https://github.com/google-deepmind/chex","docs":"https://chex.readthedocs.io/","changelog":null,"pypi":"https://pypi.org/project/chex/","npm":null,"openapi_spec":null,"status_page":null,"smithery":null,"compatibility":{"summary":{"python_range":"3.10–3.9","success_rate":30,"avg_install_s":12,"avg_import_s":2.64,"wheel_type":"wheel"},"url":"https://checklist.day/v1/registry/chex/compatibility"}}