# ============================================================
# Core
# ============================================================
numpy>=1.26.0
scipy>=1.11.0
pandas>=2.0.0
tqdm>=4.66.0
pyyaml>=6.0.1
regex>=2023.10.3
packaging>=23.2
psutil>=5.9.0
requests>=2.31.0
rich>=13.7.0
colorama>=0.4.6
typing_extensions>=4.8.0

# ============================================================
# Dataset / Data Processing
# ============================================================
datasets>=2.16.0
huggingface_hub>=0.20.0
pyarrow>=14.0.0
dill>=0.3.7
multiprocess>=0.70.15
xxhash>=3.4.1
fsspec>=2023.10.0
aiohttp>=3.9.0

# ============================================================
# Tokenizer
# ============================================================
tokenizers>=0.15.0
sentencepiece>=0.1.99
tiktoken>=0.5.2

# ============================================================
# Model / Training Utilities
# 注意：torch / torchvision / torchaudio 不放这里
# 它们由 setup_launch.py 根据 CUDA 自动安装
# ============================================================
einops>=0.7.0
safetensors>=0.4.2
accelerate>=0.26.0
transformers>=4.36.0

# ============================================================
# Logging / Visualization
# ============================================================
tensorboard>=2.15.0
tensorboardX>=2.6.2
matplotlib>=3.8.0
seaborn>=0.13.0
plotly>=5.18.0

# ============================================================
# Web / API / Local UI
# 如果你的 setup_launch.py 使用内置 http.server，不一定需要这些，
# 但多装不会影响，后续扩展 Web UI 会方便。
# ============================================================
fastapi>=0.108.0
uvicorn>=0.25.0
starlette>=0.32.0
jinja2>=3.1.2
python-multipart>=0.0.6
gradio>=4.14.0

# ============================================================
# Config / Validation
# ============================================================
pydantic>=2.5.0
omegaconf>=2.3.0

# ============================================================
# Experiment Tracking
# 如果不用 wandb，可以删掉这一行
# ============================================================
wandb>=0.16.0

# ============================================================
# Evaluation / Metrics
# ============================================================
scikit-learn>=1.3.0
evaluate>=0.4.1

# ============================================================
# File / Checkpoint Utils
# ============================================================
h5py>=3.10.0
joblib>=1.3.0

# ============================================================
# Notebook / Debug Optional
# ============================================================
ipython>=8.18.0
jupyter>=1.0.0
notebook>=7.0.0

# ============================================================
# Speed / Build Helpers
# 不包含 flash-attn / xformers，避免 Windows 或 CUDA 编译失败
# ============================================================
ninja>=1.11.1
