gymnasium~=0.29.0
wandb~=0.17.4
moviepy~=1.0.3
shimmy~=1.1.0
optree~=0.10.0
plotly~=5.20.0
numpy==1.26.4
scikit-learn~=1.4.0
cairosvg~=2.7.1
transformer-lens~=2.4.0
mamba_lens
typeguard

[dev]
notebook~=6.5.4
pyright~=1.1.353
black~=23.3.0
ruff~=0.1.14
mypy~=1.9.0
pytest~=8.1.1
pytest-cov<5,>=4
pytest-env~=1.1.2
names_generator~=0.1.0
GitPython~=3.1.41

[dev-local]
pre-commit~=3.6.0
pip-tools~=7.4.1
jinja2~=3.1.4

[torch]
torch==2.4.0
torchvision==0.19
