# Use CUDA wheels for PyTorch (cu124). Remove this line if you want CPU-only.
--extra-index-url https://download.pytorch.org/whl/cu124

# ---- Core scientific stack
numpy
scipy
scikit-learn
pandas
matplotlib
tqdm
rich
prettytable
PyYAML
requests
h5py
typing_extensions
packaging
joblib
filelock
psutil
Pillow

# ---- PyTorch stack (match major/minor versions)
torch==2.6.0
torchvision==0.21.0
torchaudio==2.6.0

# ---- Federated learning / orchestration
flwr[simulation]==1.11.1
ray==2.10.0
hydra-core==1.3.2
flwr-datasets[vision]>=0.5.0
wandb>=0.18.0
datasets==3.1.0
einops>=0.8.0
regex>=2024.9.0
sympy>=1.13.0

# ---- (Optional) Jupyter bits
ipykernel
jupyter_client
jupyter_core
traitlets
tornado
pyzmq
