MixtureOfRecursions/requirements.txt at master · Ratnaditya-J/MixtureOfRecursions · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
# Mixture-of-Recursions (MoR) Requirements
# Pinned versions for reproducibility

# Core ML and Data Science
torch==2.1.2
transformers==4.36.2
datasets==2.16.1
numpy==1.26.3
pandas==2.1.4
scikit-learn==1.3.2
accelerate==0.25.0

# Efficiency & Optimization for MoR
# flash-attn==2.4.2  # Uncomment if using Flash Attention
# triton==2.1.0      # Uncomment if using Triton kernels
# xformers==0.0.23.post1  # Uncomment if using xFormers
einops==0.7.0

# Training & Evaluation
evaluate==0.4.1
wandb==0.16.1
tensorboard==2.15.1
hydra-core==1.3.2
rich==13.7.0

# Visualization (required for analysis tools)
matplotlib==3.8.2
seaborn==0.13.1
plotly==5.17.0

# Jupyter (for interactive notebooks)
jupyter==1.0.0
ipykernel==6.28.0
ipywidgets==8.1.1

# Utilities
python-dotenv==1.0.0
tqdm==4.66.1
requests==2.31.0
pyyaml==6.0.1
click==8.1.7

# Testing
pytest==7.4.4
pytest-cov==4.1.0
pytest-mock==3.12.0

# Development
black==23.12.1
isort==5.13.2
flake8==7.0.0
pre-commit==3.6.0

# Optional: API integrations (uncomment if needed)
# openai==1.6.1
# anthropic==0.8.1