Descrição
Formas de arrays, dtypes e estruturas PyTree podem ser descritas com anotações de tipo e checagens opcionais em runtime. Desenvolvedores JAX usam jaxtyping para documentar código de modelos e encontrar erros de forma mais cedo. Checagem em runtime deve ser equilibrada com desempenho em loops de treino ou inferência.