Skip to main content

Ctrl+K

Getting Started

Installing JAX
JAX Quickstart
How to Think in JAX
🔪 JAX - The Sharp Bits 🔪
JAX Frequently Asked Questions (FAQ)

Tutorial: JAX 101

Further Resources

User Guides
Advanced Tutorials
Developer Documentation
Building on JAX
Notes
Public API: jax package

Change log
JAX Glossary of Terms

.rst

jax.tree_util.treedef_is_leaf

Contents

treedef_is_leaf()

jax.tree_util.treedef_is_leaf#

jax.tree_util.treedef_is_leaf(treedef)[source]#

Parameters: treedef (PyTreeDef) –
Return type: bool

previous

jax.tree_util.treedef_children

next

jax.tree_util.treedef_tuple

Contents

treedef_is_leaf()

By The JAX authors

© Copyright 2023, The JAX Authors. NumPy and SciPy documentation are copyright the respective authors..