Content - a9d19934052aca5e2b4a5498aeb13091417eecd0 - 177cdf6/docs/jax.tree.rst

visit type:

Tip revision: 9e62994bce7c7fcbb2f6a50c9ef89526cd2c2be6 authored by jax authors on 09 May 2024, 19:32:51 UTC
Merge pull request #21135 from hawkinsp:release

Tip revision: 9e62994

jax.tree.rst

``jax.tree`` module
===================

.. currentmodule:: jax.tree

.. automodule:: jax.tree

List of Functions
-----------------

.. autosummary::
   :toctree: _autosummary

   all
   flatten
   leaves
   map
   reduce
   structure
   transpose
   unflatten

Browse the archive

https://github.com/google/jax