apebench==0.1.1
jaxtyping
jax-tqdm
lineax
exponax
sympy
pandas
matplotlib
seaborn
jaxopt