jax>=0.9.2
jaxtyping>=0.3.9
orbax-checkpoint>=0.11.34

[rich]
plotext>=5.3.2
plotille>=6.0.5
rich>=14.3.3

[tensorboard]
torch>=2.11.0

[wandb]
wandb>=0.25.1
