Add JAX Lost Cities rules engine

2026-07-04 19:37:27 +09:00
parent 8f380928e6
commit 1d7758b7d3
7 changed files with 781 additions and 1 deletions
@@ -13,6 +13,8 @@ dependencies = [
    "pydantic>=2.7",
    "PyYAML>=6.0",
    "torch>=2.3.0",
+    "jax>=0.10.2",
+    "chex>=0.1.92",
 ]

 [project.optional-dependencies]
@@ -45,7 +47,7 @@ build-backend = "setuptools.build_meta"

 [tool.setuptools.packages.find]
 where = ["src"]
-include = ["coolrl_lost_cities*"]
+include = ["coolrl_lost_cities*", "lost_cities_jax*"]

 [tool.setuptools.package-data]
 "coolrl_lost_cities.games.classic" = [