We introduce PuzzleJAX, a benchmark for reasoning and learning. 🧩💡🦎
PuzzleJAX compiles hundreds of existing grid-based PuzzleScript games to hardware-accelerated JAX environments, and allows researchers to define new tasks via PuzzleScript's concise rewrite rule-based DSL.
26 days ago