[Feature Request] Improve PyTree Performance
Benjamin-eecs opened this issue · 0 comments
Benjamin-eecs commented
Motivation
Improve the speed of the current PyTree implemention: tree_map, tree_flatten and unflatten
Solution
An extra C++ library with python bindings
Resource
Checklist
- I have checked that there is no similar issue in the repo (required)