chex has 2.1M monthly downloads and 944 GitHub stars. Version 0.1.92 supports Python 3.11+, first published in 2020.
It is actively maintained under a permissive license, last released in Jun 2026.
It is built on top of absl-py, jax and jaxlib. It is used by 153 tracked packages, including distrax, alphafold-colabfold and dm-pix.
Package Insights
((week_daily_avg - month_daily_avg) / month_daily_avg) * 100Weekly Downloads
GitHub Stars
Downloads by OS
Python Versions
Top Countries
Dependencies
5 optional dependencies
- cloudpickle[test]
- dm-tree[test]
- sphinx[docs]
- sphinx-book-theme[docs]
- sphinxcontrib-katex[docs]
Used By
- distrax
- alphafold-colabfold
- dm-pix
- rlax
- kauldron
- eformer
- mctx
- scib-metrics
- jax-tqdm
- jax-ai-stack
- mpax
- praxis
- marin-levanter
- differt
- jumanji
- ejkernel
- esquilax
- flashbax
- alberta-framework
- jaxmarl
- benchmarkdpfair
- hyperbolix
- torax
- qdax
- gridvoting-jax
- cd-dynamax
- mbi
- fjformer
- prophetverse
- flash-hog
- xax
- xtructure
- parabellum
- metalog-jax
- dm-robotics-geometry
- jorbit
- bijx
- dm-clrs
- bacpipe
- horqrux
- biorobot
- safenax
- dfax
- alpa
- pyrovelocity
- hackable-diffusion
- juxtapose
- off
- envrax
- jaxagents