cvxpy
mosek
numpy<2
fire
tqdm
scikit-learn>=1.3
scipy
datasets
transformers
zstandard
jax
jaxlib
jaxtyping
jinja2>=3.1.2
pytest
matplotlib
jaxopt
torch
--index-url https://download.pytorch.org/whl/cu126
