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.2 source code.tar.gz	2024-09-11	758.3 kB	0
Penzai 0.2.2 source code.zip	2024-09-11	925.9 kB	0
README.md	2024-09-11	360 Bytes	0
Totals: 3 Items		1.7 MB	0

Added the ability to catch exceptions during rendering of Selection objects
Added additional methods to Penzai variables to facilitate interoperability with other JAX ecosystem libraries
Removed references to deprecated named_shape field of jax.ShapeDtypeStruct
Added citation info for Penzai/Treescope paper (https://arxiv.org/abs/2408.00211)

Source: README.md, updated 2024-09-11

Other Useful Business Software

MongoDB Atlas runs apps anywhere

Deploy in 115+ regions with the modern database for every enterprise.

MongoDB Atlas gives you the freedom to build and run modern applications anywhere—across AWS, Azure, and Google Cloud. With global availability in over 115 regions, Atlas lets you deploy close to your users, meet compliance needs, and scale with confidence across any geography.

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

MongoDB Atlas runs apps anywhere

Deploy in 115+ regions with the modern database for every enterprise.

Start 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

Penzai Files

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

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