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

↑ up