johnmarktaylor91/torchlens

Capturez chaque activation et gradient de n'importe quel modèle PyTorch — en avant et en arrière — avec la visualisation automatique du graphe, des métadonnées riches et des interventions en direct. Fonctionne sur n'importe quelle architecture, y compris dynamique et récurrente.