jax>=0.3.25
typing_extensions
pgx
