ssm-simulators>=0.12.2
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
typer>=0.9.0

[all]
mlflow>=3.6.0
huggingface-hub>=0.20.0

[hf]
huggingface-hub>=0.20.0

[mlflow]
mlflow>=3.6.0
