# ─── Core ───────────────────────────────────────────────────────────
torch>=2.1.0
torchvision>=0.16.0

# ─── HuggingFace stack ──────────────────────────────────────────────
transformers>=4.40.0
datasets>=2.18.0
tokenizers>=0.19.0
accelerate>=0.28.0
safetensors>=0.4.3
huggingface_hub>=0.22.0

# ─── Training helpers ───────────────────────────────────────────────
trl>=0.8.6           # DPO, SFT trainer wrappers
peft>=0.10.0         # LoRA / QLoRA

# ─── Mamba-2 (optional — falls back to pure PyTorch if unavailable) ─
# mamba-ssm>=2.0.4   # requires CUDA 11.8+ and a custom build
# causal-conv1d>=1.2.0

# ─── Unsloth (T4-optimised kernels — install separately in Colab) ──
# unsloth[colab-new] @ git+https://github.com/unslothai/unsloth.git

# ─── Math / utilities ───────────────────────────────────────────────
einops>=0.7.0
numpy>=1.26.0

# ─── Optional: W&B logging ──────────────────────────────────────────
# wandb>=0.17.0

# ─── Export ─────────────────────────────────────────────────────────
# llama-cpp-python>=0.2.0   # for local GGUF infeexportrence after 
