step helper

Would you have an idea how one can adapt the code to be able to do auto-regressive steps (for inference)? Is the fact that the `input_len` is fixed an issue?

Start by reading the repository's Python implementation and the code that uses input_len for inference. Determine how auto-regressive steps should work when input_len is fixed, then define a concrete completion criterion for supporting inference-time stepping and validate it with the repository's existing checks.

Written by the indexing model from the issue text.