Creative chess puzzle generation with diffusion models: new RL recipe, open weights

TZahavy · x · 2026-10-03

Eric Malmi's team released a preprint on creative chess puzzle generation using masked diffusion models. In his MSc thesis, Aatu Selkee added a new RL recipe on top of the masked diffusion model, improving on prior SOTA. The generator can be conditioned by theme, partial board position, or best move, extending the team's earlier puzzle-generation work. They are open-sourcing the model — the first open-weights models for chess puzzle generation — alongside a public demo and paper.

Original post →

More from Research

Research channel →