/torchtrail

torchtrail: trace the graph of torch functions and modules for visualization, reports, etc

Primary LanguagePythonMIT LicenseMIT