Penzai Files

A JAX research toolkit to build, edit, & visualize neural networks

This is an exact mirror of the Penzai project, hosted at https://github.com/google-deepmind/penzai. SourceForge is not affiliated with Penzai.

The interactive file manager requires Javascript. Please enable it or use sftp or scp.
You may still browse the files here.

Name	Modified	Size	InfoDownloads / Week
Parent folder
Penzai 0.2.1 source code.tar.gz	2024-08-07	757.5 kB	0
Penzai 0.2.1 source code.zip	2024-08-07	925.1 kB	0
README.md	2024-08-07	224 Bytes	0
Totals: 3 Items		1.7 MB	0

Adjusted colors for TransformerBlock and Attention when rendered with Treescope.
Fixed an outdated link in the README.

Source: README.md, updated 2024-08-07

Other Useful Business Software

Custom VMs From 1 to 96 vCPUs With 99.95% Uptime

General-purpose, compute-optimized, or GPU/TPU-accelerated. Built to your exact specs.

Live migration and automatic failover keep workloads online through maintenance. One free e2-micro VM every month.

Try Free

8 Monitoring Tools in One APM. Install in 5 Minutes.

Errors, performance, logs, uptime, hosts, anomalies, dashboards, and check-ins. One interface.

AppSignal works out of the box for Ruby, Elixir, Node.js, Python, and more. 30-day free trial, no credit card required.

Start Free

Custom VMs From 1 to 96 vCPUs With 99.95% Uptime

General-purpose, compute-optimized, or GPU/TPU-accelerated. Built to your exact specs.

Live migration and automatic failover keep workloads online through maintenance. One free e2-micro VM every month.

Try Free

Recommended Projects

Google DeepMind GraphCast and GenCast
Global weather forecasting model using graph neural networks and JAX
Materials Discovery: GNoME
AI discovers 520000 stable inorganic crystal structures for research
Lucid
A collection of infrastructure and tools for research
JAX Toolbox
Public CI, Docker images for popular JAX libraries
Hummingbird
Hummingbird compiles trained ML models into tensor computation