# Base, portable dependencies for the ARC 2025 solution.
# Install these for local development or open-source usage.

# Core ML stack (CPU-friendly defaults)
torch==2.8.0
torchvision==0.23.0

# Hugging Face ecosystem
transformers==4.57.0
accelerate==1.10.1
peft==0.17.1
datasets==4.1.1
huggingface-hub==0.35.3
safetensors==0.6.2
tokenizers==0.22.1
sentencepiece==0.2.1
bitsandbytes==0.48.1
matplotlib==3.10.6

# Scientific Python
numpy==2.3.3
pandas==2.3.3
scikit-learn==1.7.2
pyarrow==21.0.0
psutil==7.1.0
protobuf==6.32.1

# JAX / Flax components used for TPU/GPU flows
jax==0.7.2
jaxlib==0.7.2
flax==0.12.0
optax==0.2.6

# Utilities
tqdm==4.67.1
tomli; python_version < "3.11"

muon-optimizer==0.1.0

statsmodels==0.14.5

# Cluster-specific binaries live in requirements.cluster.txt
