torch>=2.0.0
transformers>=4.30.0
numpy>=1.21.0
pandas>=1.3.0

[all]
psidyn[dev,experiments,quantization]

[dev]
pytest>=7.0.0
pytest-cov>=4.0.0
ruff>=0.1.0
mypy>=1.0.0

[experiments]
sentence-transformers>=2.2.0
scipy>=1.7.0
tqdm>=4.60.0

[quantization]
bitsandbytes>=0.41.0
accelerate>=0.20.0
