jax>=0.3.4
numpy>=1.20.0
typeguard>=2.13.3
typing_extensions>=4.2.0
