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

[all]
nnsight[test]

[test]
pytest>=6.0
pytest-cov
