opendilab/DI-treetensor

DI-treetensor simplifies managing and performing calculations on data structured like a tree, where different parts can have their own properties. It takes in complex, hierarchical data and lets you apply mathematical operations or manipulations across the entire structure, outputting the transformed tree. This is for machine learning engineers and researchers who work with deep learning models involving nested data structures, like those in reinforcement learning or complex neural network architectures.

213 stars. Used by 1 other package. No commits in the last 6 months. Available on PyPI.