matplotlib~=3.4.3
numpy~=1.21.2
setuptools~=59.5.0
torch~=1.10.2
sklearn~=0.0
scikit-learn~=1.0.1
wandb~=0.12.10
jax~=0.2.21
jaxlib~=0.1.75
distrax~=0.1.0
scipy~=1.7.3
pytest~=6.2.5
torchvision~=0.11.3
disent~=0.3.3
dm-env~=1.5
PyYAML~=6.0
six~=1.16.0
Pillow~=8.3.2
imageio~=2.13.3
pytorch-lightning~=1.5.9
jsonargparse[signatures]~=4.1.4
hydra-core~=1.1.1
pre-commit
omegaconf~=2.1.1
dm-haiku # @ git+https://github.com/deepmind/dm-haiku@b2ae68ecbec47e75f12eb318c40b0b9cb61b5bce
tueplots