torch>=2.0.0
numpy>=1.24.0
scipy>=1.10.0
pyyaml>=6.0

[all]
kinetic-ai[dev,experiments,serve]

[data]
datasets>=2.14
tokenizers>=0.15
transformers>=4.35

[dev]
pytest>=7.0
pytest-cov>=4.0
types-PyYAML
types-requests
mypy>=1.0
ruff>=0.1.0

[experiments]
matplotlib>=3.7
pandas>=2.0
tqdm>=4.65
tensorboard>=2.13

[serve]
fastapi>=0.100.0
uvicorn>=0.24.0
httpx>=0.25.0
requests>=2.31.0
huggingface_hub>=0.20
