Skip to main content

Ctrl+K

penzai

Guides

How to Think in Penzai
Pretty-Printing With Treescope
Visualizing NDArrays with Treescope
Selectors and Selections
Named Axes in Penzai
Data Effects - Scoped Side Effects and State in Penzai Models
Jitting and Sharding Penzai Models

Example Notebooks

LoRA From Scratch - Patching Pretrained Models in Penzai
Induction Heads in Gemma 7B
Gemma From Scratch: Inspecting and Re-implementing Gemma with Penzai

API Reference

pz: Penzai’s alias namespace
penzai.core
penzai.nn
penzai.data_effects
penzai.treescope

Extras

penzai.toolshed
penzai.example_models
- gemma
- simple_mlp
  - DropoutMLP
  - MLP

Repository
Open issue

.rst

build_copy_button

Contents

build_copy_button()

build_copy_button#

penzai.treescope.foldable_representation.common_structures.build_copy_button(path: tuple[Any, ...] | None) → RenderableTreePart[source]#

previous

EmptyWithHeightGuess

next

build_custom_foldable_tree_node

Contents

build_copy_button()

By the Penzai Authors

© Copyright 2024, The Penzai Authors.