Topic: "tree-flatten"
metaopt/optree
OpTree: Optimized PyTree Utilities
Language: Python - Size: 1.15 MB - Last synced at: 1 day ago - Pushed at: 1 day ago - Stars: 183 - Forks: 10
Related Topics
OpTree: Optimized PyTree Utilities
Language: Python - Size: 1.15 MB - Last synced at: 1 day ago - Pushed at: 1 day ago - Stars: 183 - Forks: 10