Skip to main content

Ctrl+K

penzai

Core Tools

Pretty-Printing With Treescope
Visualizing NDArrays with Treescope
Selectors and Selections
Named Axes in Penzai

Neural Network Guides (V1 API)

How to Think in Penzai
Data Effects - Scoped Side Effects and State in Penzai Models
Jitting and Sharding Penzai Models
LoRA From Scratch - Patching Pretrained Models in Penzai
Induction Heads in Gemma 7B
Gemma From Scratch: Inspecting and Re-implementing Gemma with Penzai

Neural Network Guides (V2 API)

How to Think in Penzai (v2 NN API)
Changes in the V2 API
How-To Guide - Common Tasks
Jitting and Sharding Penzai Models (V2 API)
LoRA From Scratch - Patching Pretrained Models in Penzai (V2 API)
Induction Heads in Gemma 7B (V2 API)

API Reference

pz: Penzai’s alias namespace
penzai.core
penzai.nn
penzai.data_effects
penzai.treescope: Penzai’s interactive pretty-printer
penzai.toolshed
penzai.example_models
- gemma
- simple_mlp
  - DropoutMLP
  - MLP
Simplified V2 API (penzai.experimental.v2)

Repository
Open issue

Search

Ctrl+K

By the Penzai Authors

© Copyright 2024, The Penzai Authors.