torch==2.11.0
torchvision
transformers<6,>=5.15
datasets>=2.14.0
accelerate>=0.24.0
peft>=0.6.0
torchao<0.18,>=0.17.0
bitsandbytes>=0.41.0
compressed-tensors>=0.15.0
numpy>=1.21.0
pydantic>=2.0.0
pyyaml>=6.0
huggingface-hub>=0.17.0
tqdm>=4.64.0
psutil>=5.9.0
litellm
asteval
langchain
langchain-community
pandas>=1.3.0
scikit-learn>=1.0.0
click>=8.0.0
rich>=12.0.0
typer>=0.7.0
trl==1.7.1
langgraph>=0.2
openenv>=0.3.1
httpx>=0.28.1
websockets>=15.0.1
colorama>=0.4.6
requests>=2.28.0
wandb>=0.15.0
tensorboard>=2.12.0
matplotlib>=3.5.0
pillow<12,>=11.3.0
evaluate>=0.4.0
rouge-score>=0.1.2
lm-eval>=0.4.0
nltk>=3.8.0
sentence-transformers>=2.2.0
chromadb>=0.5.0
langchain-text-splitters>=0.3.0
beautifulsoup4>=4.12.0
python-docx>=1.0.0
python-pptx>=0.6.23
pdfplumber<0.11.10,>=0.10.0
pyarrow>=14.0.0
networkx>=3.0
groq>=0.9.0
jsonschema>=4.0.0
codebleu>=0.1.0
bert-score>=0.3.13
psycopg2-binary>=2.9.0

[dev]
pytest>=7.0.0
pytest-asyncio>=0.21.0
pytest-cov>=4.0.0
pytest-xdist>=3.0.0
pytest-mock>=3.10.0
pytest-timeout>=2.1.0
hypothesis>=6.70.0
black>=23.0.0
isort>=5.12.0
ruff>=0.4.0
mypy>=1.5.0
bandit>=1.7.0
pre-commit>=3.0.0

[docs]
mkdocs<2,>=1.6
mkdocs-material<10,>=9.5
pymdown-extensions<11,>=10

[flash-attn]
flash-attn>=2.5.0

[service]
fastapi>=0.110.0
uvicorn>=0.29.0
websockets>=15.0.1

[vllm]
trl[vllm]==1.7.1
