numpy>=1.20.0
pandas>=1.3.0
matplotlib>=3.4.0
plotly>=5.0.0
torch>=1.9.0
psutil>=5.8.0
gputil>=1.4.0
py-cpuinfo>=8.0.0
pynvml>=11.0.0
tqdm>=4.60.0
dash>=2.0.0
seaborn>=0.11.0
scikit-learn>=1.0.0
scipy>=1.7.0

[advanced]
scikit-learn>=1.0.0
scipy>=1.7.0
networkx>=2.6.0
statsmodels>=0.13.0
umap-learn>=0.5.0
hdbscan>=0.8.0
pyro-ppl>=1.7.0
pygraphviz>=1.7
qiskit>=0.30.0
pennylane>=0.20.0
torchviz>=0.0.2
optuna>=2.10.0
ray[tune]>=1.13.0
tensorboardX>=2.5.0

[all]
pytorch-lightning>=1.5.0
transformers>=4.5.0
deepspeed>=0.5.0
openai>=0.27.0
anthropic>=0.3.0
bokeh>=2.4.0
networkx>=2.6.0
statsmodels>=0.13.0
umap-learn>=0.5.0
hdbscan>=0.8.0

[deepspeed]
deepspeed>=0.5.0

[dev]
pytest>=6.0.0
pytest-cov>=2.12.0
black>=21.5b2
isort>=5.9.0
mypy>=0.812
flake8>=3.9.0
sphinx>=4.0.0
sphinx-rtd-theme>=0.5.0
twine>=3.4.0
build>=0.7.0

[huggingface]
transformers>=4.5.0

[lightning]
pytorch-lightning>=1.5.0

[llm]
transformers>=4.5.0
openai>=0.27.0
anthropic>=0.3.0

[viz]
dash>=2.0.0
seaborn>=0.11.0
bokeh>=2.4.0
