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

random_stream

random_stream#

A simple random stream abstraction.

Classes

Helper object to construct a stream of random numbers.

previous

MisconfiguredLayerError

next

RandomStream

By the Penzai Authors

© Copyright 2024, The Penzai Authors.