refactor(autogram): Type `args` as `tuple[PyTree, ...]`#439

Closed

PierreQuinton wants to merge 2 commits intomainfrom

Commits on Oct 1, 2025

Type `args` as `tuple[PyTree, ...]` to highlight both the fact that it is a tuple and has recursive structure.
PierreQuinton
committed
Type `in_dims` as `tuple[PyTree, ...]` to highlight both the fact that it is a tuple and has recursive structure.
PierreQuinton
committed