new trainer image

2026-05-31 08:33:36 +00:00 · 2026-02-19 13:03:25 +01:00
parent 843564eeb0
commit 5912062dc0
3 changed files with 79 additions and 0 deletions
--- a/docker/trainer.requirements.txt
+++ b/docker/trainer.requirements.txt
@@ -0,0 +1,13 @@
+numpy>=1.24.0
+pandas>=2.0.0
+scipy>=1.11.0
+gymnasium>=0.29.0
+stable-baselines3>=2.2.0
+tensorboard>=2.15.0
+wandb>=0.17.0
+tensorflow-probability==0.24.0
+flax==0.10.7
+optax==0.2.7
+distrax==0.1.5
+orbax-checkpoint==0.11.32
+chex==0.1.90