torch>=2.0.0
transformers>=4.30.0
numpy>=1.24.0
tqdm>=4.65.0
datasets>=2.12.0
accelerate>=0.20.0
safetensors>=0.3.1
einops>=0.6.1
mamba-ssm>=1.0.0

[all]
capibara-gpt[training]
capibara-gpt[quantization]
capibara-gpt[monitoring]

[monitoring]
psutil>=5.9.0
prometheus-client>=0.17.0

[quantization]
bitsandbytes>=0.41.0
optimum>=1.12.0

[training]
wandb>=0.15.0
tensorboard>=2.13.0
pytorch-lightning>=2.0.0
