a :gh¼ã@s0dZddlmZmZmZmZmZmZmZm Z dS)z˜Utilities for working with tree-like container data structures. The :mod:`jax.tree` namespace contains aliases of utilities from :mod:`jax.tree_util`. é)ÚallÚflattenÚleavesÚmapÚreduceÚ structureÚ transposeÚ unflattenN) Ú__doc__Z jax._src.treerrrrrrrr ©r r úX/mounts/lovelace/software/anaconda3/envs/metaDMG/lib/python3.9/site-packages/jax/tree.pyÚs