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

[all]
nnsight[diffusers,test,vllm]

[diffusers]
diffusers

[test]
pytest>=6.0
pytest-cov

[vllm]
vllm==0.15.1
triton==3.5.0
