{"library":"chex","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"]}]}