datasets>=3.2.0
einops>=0.8.0
jsonlines>=4.0.0
nnsight<0.4.0,>=0.3.7
onnx2torch>=1.5.15
python-chess>=1.999
scikit-learn>=1.6.1
tensordict>=0.6.2
typing-extensions>=4.12.2

[viz]
matplotlib>=3.10.0
