Ctrl+K

penzai

Neural Networks

JAX Tools

See Also

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)

Search

Ctrl+K

By the Penzai Authors