ssm-simulators>=0.10.0
scipy>=1.15.2
pandas>=2.2.3
torch>=2.7.0
flax>=0.10.6
tqdm>=4.67.1
frozendict>=2.4.6
onnx>=1.17.0
matplotlib>=3.10.1

[viz]
wandb>=0.19.11
