# tested with torch 2.11, transformers 5.8.1, flash-linear-attention 0.5.2, causal-conv1d 1.7.0, peft 0.20 on H200
torch>=2.8
transformers>=5.8
huggingface_hub>=0.34
safetensors>=0.4
# image input: the checkpoints' image processor uses torchvision; install the build that matches your torch
pillow>=10
torchvision>=0.23
# required on CUDA: without them transformers silently runs a plain torch path (>10x slower)
flash-linear-attention>=0.5.2; sys_platform == "linux"
causal-conv1d>=1.7; sys_platform == "linux"          # CUDA build; if pip compiles it, add --no-build-isolation
# Apple Silicon: the forward pass runs in MLX instead (startlux_decision/mlx_model.py)
mlx-lm>=0.31.3; sys_platform == "darwin" and platform_machine == "arm64"
# fine-tuning
peft>=0.20
# evaluation (Typed Decisions parquet); the Decision Index kit is installed from its own repository
pyarrow>=15