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

array_autovisualizer

array_autovisualizer#

An automatic NDArray visualizer using arrayviz.

Classes

ArrayAutovisualizer

An automatic visualizer for arrays.

previous

arrayviz

next

ArrayAutovisualizer

By the Penzai Authors

© Copyright 2024, The Penzai Authors.