jax>=0.4.33
jaxlib>=0.4.33
jaxtyping==0.2.34
beartype==0.18.5
