Sam Earle
@smearle.bsky.social
PhD at NYU in AI and Games. RL, environment generation, open-ended learning.
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.