Descrição
Estado de treino JAX e dados de modelos podem ser salvos e restaurados com utilitários compartilhados de checkpoint. Engenheiros de ML usam Orbax Checkpoint em treinos longos, recuperação de experimentos e fluxos distribuídos de modelos. Checkpoints podem conter pesos de modelos, estado de otimizadores e informações aprendidas sensíveis.