Metadata-Version: 2.4
Name: imt-tree-utils
Version: 0.2.7
Summary: Utilities for working with Pytrees
Author-email: Simon Bachhuber <simon.bachhuber@fau.de>
Project-URL: Homepage, https://github.com/SimiPixel/tree_utils
Classifier: Programming Language :: Python :: 3
Classifier: License :: OSI Approved :: MIT License
Classifier: Operating System :: OS Independent
Requires-Python: >=3.10
Description-Content-Type: text/markdown
Requires-Dist: jax
Requires-Dist: jaxlib
Requires-Dist: dm-tree

A collection of useful functions when working with PyTrees.

Some of them are copied from `deepmind/acme` but having `acme` felt like a heavy dependency for these few lines of code..
