chore: release v0.2.0

2026-02-10 08:08:11 +00:00
parent dbfa9e31fa
commit b568ac5600
123 changed files with 9732 additions and 497 deletions
--- a/docker/docker-compose.yml
+++ b/docker/docker-compose.yml
@@ -0,0 +1,41 @@
+volumes:
+  motrixlab-data:
+    name: motrixlab-data
+
+services:
+  motrixlab-training:
+    image: motphys/motrixlab:latest
+    container_name: motrixlab-training
+    volumes:
+      - motrixlab-data:/root/motrixlab/runs
+    deploy:
+      resources:
+        reservations:
+          devices:
+            - driver: nvidia
+              count: all
+              capabilities: [gpu]
+    command:
+      - scripts/train.py
+      - --train-backend
+      - ${MOTRIX_TRAIN_BACKEND:-jax}
+      - --num-envs
+      - ${MOTRIX_NUM_ENVS:-2048}
+      - --env
+      - ${MOTRIX_ENV:-cartpole}
+    restart: no
+
+  motrixlab-tensorboard:
+    image: motphys/motrixlab:latest
+    container_name: motrixlab-tensorboard
+    ports:
+      - "6006:6006"
+    volumes:
+      - motrixlab-data:/root/motrixlab/runs:ro
+    command:
+      - tensorboard
+      - --host
+      - 0.0.0.0
+      - --logdir
+      - runs/${MOTRIX_ENV:-cartpole}
+    restart: unless-stopped