Skip to main content
Ctrl+K
Logo image

User guides

  • 🏃 Getting started
  • 📜 Stateful computations
  • 🥶 Dealing with non-jax types
  • 🍳 Common recipes

Examples

  • 🧮 Building a mini optimizer library

API Documentation

  • pytreeclass API
    • 🎯 Core API
    • 💼 Masking API
    • ➕ Advanced API
    • 🎨 Pretty printing API
  • .rst

pytreeclass API

pytreeclass API#

API Documentation

  • 🎯 Core API
    • TreeClass
    • autoinit()
    • leafwise()
    • is_tree_equal()
    • field()
    • fields()
  • 💼 Masking API
    • is_nondiff()
    • freeze()
    • unfreeze()
    • is_frozen()
    • tree_mask()
    • tree_unmask()
  • ➕ Advanced API
    • bcmap()
    • Partial
    • AtIndexer
    • BaseKey
  • 🎨 Pretty printing API
    • tree_diagram()
    • tree_graph()
    • tree_mermaid()
    • tree_repr()
    • tree_str()
    • tree_summary()

previous

🧮 Building a mini optimizer library

next

🎯 Core API

By Mahmoud Asem

© Copyright 2023, Mahmoud Asem.