Descrição
Fornece utilitários PyTree otimizados para trabalhar com estruturas Python aninhadas usadas em bibliotecas numéricas e de aprendizado de máquina. Ajuda a achatar, mapear e transformar árvores estruturadas de parâmetros com eficiência.
Transformações PyTree podem afetar parâmetros de modelos ou estado de treinamento. Desenvolvedores devem testar o tratamento da estrutura com cuidado para manter valores alinhados aos campos pretendidos do modelo.