pytrees
last updated: Oct 20, 2023
https://jax.readthedocs.io/en/latest/pytrees.html
A neat idea to treat any container type as a tree node, and any non-container as a leaf node; then you can define operations on "pytrees" with respect to the leafs of heterogenous objects.
via news.yc