Skip to main content

Ctrl+K

penzai

Neural Networks

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

JAX Tools

Selectors and Selections
Named Axes in Penzai

See Also

Pretty-Printing With Treescope
Array Visualization With Treescope

API Reference

pz: Penzai’s alias namespace
- pz.nn: Neural network alias namespace
- pz.ts: Treescope alias namespace
penzai.core
penzai.nn
penzai.models
- simple_mlp
  - DropoutMLP
  - MLP
- transformer
penzai.toolshed
penzai.treescope (Moved!)
penzai.deprecated.v1 (Original V1 API)

Repository
Open issue

.rst

simple_mlp

simple_mlp#

A simple multi-layer perceptron.

Classes

`DropoutMLP`	Sequence of Affine layers with dropout.
`MLP`	Sequence of Affine layers.

previous

temperature_sample_pyloop

next

DropoutMLP

By the Penzai Authors

© Copyright 2024, The Penzai Authors.