
[ace]
numpy<3,>=1.26
openai<3,>=1.50
python-dotenv<2,>=1.2
tiktoken<1,>=0.12

[acon]
requests<3,>=2.32
jinja2<4,>=3.1
tiktoken<1,>=0.9

[acon-fit]
requests<3,>=2.32
jinja2<4,>=3.1
tiktoken<1,>=0.9
numpy<3,>=1.26
openai<3,>=1.50
PyYAML<7,>=6
msal<2,>=1.30
tqdm<5,>=4.66

[adaptive-rag-fit]
torch==1.13.0
transformers==4.28.1
tokenizers==0.13.3
accelerate==0.15.0
numpy==1.26.4
datasets==2.14.7
pyarrow==14.0.2
fsspec==2023.10.0
huggingface-hub==0.19.4
nltk==3.9.2
safetensors==0.5.3
sentencepiece==0.2.1
jsonlines==4.0.0

[benchmarks]
pyarrow<24,>=18

[clariq]
pandas<3,>=2.1
scikit-learn<2,>=1.4

[compression]
llmlingua==0.2.2

[decomp]
torch==1.13.0
transformers==4.24.0
numpy==1.26.4
jsonnet==0.21.0
nltk==3.9.2
openai==0.28.1
diskcache==5.6.3
python-dateutil==2.9.0.post0
requests==2.32.5

[dense-retrieval]
sentence-transformers<6,>=3.0
transformers<6,>=4.51

[dev]
build>=1.2
coverage[toml]>=7.6
pytest>=8.3
pytest-cov>=5.0
ir-measures<0.5,>=0.4.1
pyarrow<24,>=18
requests<3,>=2.32
rouge-score<0.2,>=0.1.2
ruff==0.12.5
scikit-learn<2,>=1.4

[evaluation]
ir-measures<0.5,>=0.4.1
rouge-score<0.2,>=0.1.2

[gpu-monitoring]
psutil>=5.9
nvidia-ml-py>=12

[hyde-native]
numpy<3,>=1.26

[infocqr-student]
torch<3,>=2.6
transformers<5,>=4.51
sentencepiece<0.3,>=0.2
safetensors<1,>=0.5

[ircot]
numpy==1.26.4
jsonnet==0.21.0
transformers==4.24.0
spacy==3.4.1
openai==0.28.1
diskcache==5.6.3
nltk==3.9.2
requests==2.32.5
rapidfuzz<4,>=3
ftfy<7,>=6
python-dateutil<3,>=2.8
tqdm<5,>=4.66

[llmcompiler]
langchain==0.2.6
langchain-community==0.2.6
openai==1.3.7
tiktoken==0.5.2

[llmlingua2-fit]
llmlingua==0.2.2
torch<3,>=2.6
transformers<6,>=4.51
spacy==3.8.7
tensorboard==2.20.0
numpy<2,>=1.26
scikit-learn<2,>=1.4
safetensors<1,>=0.5

[local]
torch>=2.2
transformers>=4.40

[longbench]
jieba==0.42.1
rouge==1.0.1
fuzzywuzzy==0.18.0

[mce-text]
openai<3,>=1.50
httpx<1,>=0.27
python-dotenv<2,>=1.2

[paper]
matplotlib<4,>=3.8
numpy<3,>=1.26
pandas<3,>=2.1
scikit-learn<2,>=1.4

[plan-and-budget]
numpy==1.26.4

[process-monitoring]
psutil>=5.9

[qrag]
datasets<5,>=3
hydra-core<2,>=1.3
nltk<4,>=3.9
numpy<3,>=1.26
omegaconf<3,>=2.3
rotary-embedding-torch<1,>=0.8
sortedcontainers<3,>=2.4
torch>=2.2
transformers<6,>=4.51

[qrag-fit]
datasets<5,>=3
hydra-core<2,>=1.3
nltk<4,>=3.9
numpy<3,>=1.26
omegaconf<3,>=2.3
rotary-embedding-torch<1,>=0.8
sortedcontainers<3,>=2.4
torch<3,>=2.6
transformers<6,>=4.51
safetensors<1,>=0.5
tensorboard<3,>=2.18

[querygym]
PyYAML>=6.0
tqdm>=4.66

[r2ag-fit]
torch<3,>=2.6
transformers<6,>=5.15
sentence-transformers<6,>=5
numpy<2,>=1.26
accelerate<2,>=1.10
bitsandbytes==0.49.2
safetensors<1,>=0.5
kernels==0.16.0

[recomp-fit]
torch==1.13.0
torchvision==0.14.0
sentence-transformers==2.2.2
transformers==4.24.0
tokenizers==0.13.2
huggingface-hub==0.19.4
numpy==1.26.4
pandas==1.5.3
safetensors==0.5.3

[recomp-seq2seq-fit]
torch==1.13.0
transformers==4.24.0
tokenizers==0.13.2
numpy==1.26.4
datasets==2.14.7
pyarrow==14.0.2
fsspec==2023.10.0
huggingface-hub==0.19.4
nltk==3.9.2
rouge==1.0.1
safetensors==0.5.3
sentencepiece==0.2.1

[reformer]
pandas<3,>=2.1

[reformer-fit]
numpy<3,>=1.26
pandas<3,>=2.1
requests<3,>=2.32
tqdm<5,>=4.66

[selectors]
scikit-learn<2,>=1.4

[smartsearch-grpo]
torch==2.6.0
torchvision==0.21.0
torchaudio==2.6.0
transformers==4.51.1
vllm==0.8.4
ray[default]==2.47.1
tensordict==0.6.2
hydra-core==1.3.2
omegaconf==2.3.0
antlr4-python3-runtime==4.9.3
numpy==1.26.4
torchdata==0.11.0
datasets==4.0.0
pyarrow==21.0.0
pandas==2.3.2
accelerate==1.9.0
codetiming==1.4.0
einops==0.8.1
safetensors==0.6.2
tqdm==4.67.1
requests==2.32.5
PyYAML==6.0.2
packaging==25.0
ninja==1.13.0
wheel==0.45.1
jieba==0.42.1
langid==1.1.6

[smartsearch-policy]
torch==2.6.0
torchvision==0.21.0
torchaudio==2.6.0
transformers==4.51.1
datasets==4.0.0
accelerate==1.9.0
peft==0.17.1
trl==0.9.6
torchdata==0.11.0
tyro==0.8.14
numpy==1.26.4
pyarrow==21.0.0
omegaconf==2.3.0
safetensors==0.6.2
sentencepiece==0.2.1
PyYAML==6.0.2
einops==0.8.1
matplotlib==3.10.6
nltk==3.9.1
jieba==0.42.1
rouge-chinese==1.0.3
protobuf==5.29.5
tiktoken==0.9.0
fire==0.7.1
wheel==0.45.1
ninja==1.13.0

[smartsearch-prepare]
requests<3,>=2.32
tqdm<5,>=4.66

[smartsearch-sample]
jieba==0.42.1
langid==1.1.6
numpy<2,>=1.26

[yaml]
PyYAML>=6.0
