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.5 source code.tar.gz	2025-04-08	905.2 kB	0
Penzai 0.2.5 source code.zip	2025-04-08	1.1 MB	0
README.md	2025-04-08	79 Bytes	0
Totals: 3 Items		2.0 MB	0

Bug fixes

Fixed a broken test when running with Treescope 0.1.9 (#108)

Source: README.md, updated 2025-04-08

Other Useful Business Software

Full-stack observability with actually useful AI | Grafana Cloud

Our generous forever free tier includes the full platform, including the AI Assistant, for 3 users with 10k metrics, 50GB logs, and 50GB traces.

Built on open standards like Prometheus and OpenTelemetry, Grafana Cloud includes Kubernetes Monitoring, Application Observability, Incident Response, plus the AI-powered Grafana Assistant. Get started with our generous free tier today.

Create free account

Train ML Models With SQL You Already Know

BigQuery automates data prep, analysis, and predictions with built-in AI assistance.

Build and deploy ML models using familiar SQL. Automate data prep with built-in Gemini. Query 1 TB and store 10 GB free monthly.

Try Free

Full-stack observability with actually useful AI | Grafana Cloud

Our generous forever free tier includes the full platform, including the AI Assistant, for 3 users with 10k metrics, 50GB logs, and 50GB traces.

Create free account

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

Penzai Files

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

Get an email when there's a new version of Penzai

Bug fixes