Zehua Jiang
@zehuajiang.bsky.social
PhD student at New York University; Game AI
Reposted by Zehua Jiang
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.
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.
August 27, 2025 at 11:34 PM
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.
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.