transformers
astor
cloudpickle
httpx
python-socketio[client]
pydantic>=2.9.0
torch>=2.4.0
accelerate
toml
ipython
rich

[all]
nnsight[test,vllm]

[test]
pytest>=6.0
pytest-cov

[vllm]
vllm>=0.12
triton==3.5.0
