numpy
pandas
scikit-learn
nvidia-ml-py
pydantic
rich

[dev]
ruff
black==22.6.0
pytest==7.3.2
pytest-mock==3.10.0
pytest-xdist==3.3.1
torch==2.0.1

[lint]
ruff
black==22.6.0

[test]
pytest==7.3.2
pytest-mock==3.10.0
pytest-xdist==3.3.1

[torch]
torch==2.0.1
