mirror of
https://github.com/NousResearch/atropos.git
synced 2026-04-24 17:04:55 +00:00
75 lines
No EOL
1.1 KiB
Text
75 lines
No EOL
1.1 KiB
Text
# Core dependencies
|
|
numpy>=1.24.0
|
|
scipy>=1.10.0
|
|
matplotlib>=3.7.0
|
|
seaborn>=0.12.0
|
|
pandas>=2.0.0
|
|
pyyaml>=6.0
|
|
tqdm>=4.65.0
|
|
|
|
# ML and LLM dependencies
|
|
torch>=2.0.0
|
|
transformers>=4.30.0
|
|
accelerate>=0.20.0
|
|
einops>=0.6.0
|
|
sentencepiece>=0.1.99
|
|
tiktoken>=0.4.0
|
|
openai>=0.27.0
|
|
anthropic>=0.5.0
|
|
langchain>=0.0.267
|
|
vllm>=0.1.4
|
|
|
|
# Visualization and UI
|
|
plotly>=5.14.0
|
|
gradio>=3.40.0
|
|
wandb>=0.15.0
|
|
ipywidgets>=8.0.6
|
|
ipython>=8.12.0
|
|
jupyter>=1.0.0
|
|
ipykernel>=6.22.0
|
|
imageio>=2.31.0
|
|
|
|
# Environment and utils
|
|
gymnasium>=0.28.1
|
|
pydantic>=2.0.2
|
|
python-dotenv>=1.0.0
|
|
requests>=2.31.0
|
|
aiohttp>=3.8.4
|
|
fastapi>=0.97.0
|
|
uvicorn>=0.22.0
|
|
|
|
# Testing
|
|
pytest>=7.3.1
|
|
pytest-asyncio>=0.21.0
|
|
|
|
# Atropos-specific dependencies
|
|
protobuf>=4.23.3
|
|
jax>=0.4.13
|
|
jaxlib>=0.4.13
|
|
flax>=0.7.0
|
|
optax>=0.1.5
|
|
dm-haiku>=0.0.10
|
|
brax>=0.9.0
|
|
orjson>=3.9.1
|
|
tensorboard>=2.13.0
|
|
tensorboardX>=2.6.0
|
|
cupy-cuda11x>=12.0.0; platform_system != "Darwin"
|
|
rouge-score>=0.1.2
|
|
sacrebleu>=2.3.1
|
|
datasets>=2.13.0
|
|
nltk>=3.8.1
|
|
deepspeed>=0.9.5
|
|
rich>=13.4.2
|
|
sh>=2.0.4
|
|
safetensors>=0.3.1
|
|
|
|
# HTML and visualization
|
|
jinja2>=3.1.2
|
|
markdown>=3.4.3
|
|
beautifulsoup4>=4.12.2
|
|
html5lib>=1.1
|
|
|
|
# For WebUI
|
|
flask>=2.3.2
|
|
pillow>=9.5.0
|
|
websockets>=11.0.3 |