Add JAX PPO static-opponent trainer

2026-07-04 22:30:16 +09:00
parent 768f74693d
commit 9e27f42f27
12 changed files with 1682 additions and 0 deletions
@@ -15,6 +15,9 @@ dependencies = [
    "torch>=2.3.0",
    "jax>=0.10.2",
    "chex>=0.1.92",
+    "flax>=0.12.7",
+    "optax>=0.2.8",
+    "orbax-checkpoint>=0.12.1",
 ]

 [project.optional-dependencies]
@@ -32,6 +35,7 @@ lost-cities-eval = "coolrl_lost_cities.games.classic.evaluation:main"
 lost-cities-classic-gui = "coolrl_lost_cities.games.classic.pygame_pvp:main"
 lost-cities-deep-cfr = "coolrl_lost_cities.games.classic.deep_cfr.cli:main"
 lost-cities-ismcts = "coolrl_lost_cities.games.classic.ismcts.cli:main"
+lost-cities-jax-ppo = "lost_cities_jax.ppo_cli:main"

 [dependency-groups]
 dev = [