hydra-core>=1.2.0
wandb>=0.13
numpy
torch>=1.10
transformers>=4.22.0
tokenizers
requests
Flask
scikit-learn
matplotlib
langchain==0.0.225
typing_extensions==4.5
openai
plotly

[benchmarks]
pygraphviz
graphviz
openai

[dev]
black
isort
flake8
flake8-pyproject
pydocstyle
mypy
pre-commit
pytest
pytest-cov

[docs]
sphinx==5.3.0
sphinx_rtd_theme
sphinx_autodoc_typehints

[notebook]
ipython

[sodaracer]
swig>=4.1.0
box2d-py==2.3.8
pygame

[triton]
tritonclient[all]
