# Core dependencies for minimal_export
numpy>=1.21
jax==0.4.30
jaxlib==0.4.30
flax==0.8.5
optax>=0.1
chex>=0.1
orbax-checkpoint>=0.4
omegaconf>=2.3
matplotlib>=3.5
tqdm>=4.60
PyYAML>=6.0
flashbax>=0.1
torch>=1.8
tensorboard>=2.5 
scikit-learn>=1.3
navix==0.7
seaborn>=0.12