jax>=0.4.13
jaxtyping>=0.2.20
equinox>=0.10.7
typing_extensions>=4.5.0
tensorflow-probability>=0.20.1
plum-dispatch>=2.1.0
