xlm.tasks.maze
Data preprocessing for the synthetic maze-planning task.
preprocess_fn(example, tokenizer)
Tokenize a maze path and its ordered conditioning subgoals.
on_the_fly_fn(example, tokenizer, block_size=None)
Expose model-facing path, prompt, and anchor-position fields.