This is the Linux app named Penzai whose latest release can be downloaded as Penzai0.2.5sourcecode.tar.gz. It can be run online in the free hosting provider OnWorks for workstations.
Download and run online this app named Penzai with OnWorks for free.
Follow these instructions in order to run this app:
- 1. Downloaded this application in your PC.
- 2. Enter in our file manager https://www.onworks.net/myfiles.php?username=XXXXX with the username that you want.
- 3. Upload this application in such filemanager.
- 4. Start the OnWorks Linux online or Windows online emulator or MACOS online emulator from this website.
- 5. From the OnWorks Linux OS you have just started, goto our file manager https://www.onworks.net/myfiles.php?username=XXXXX with the username that you want.
- 6. Download the application, install it and run it.
SCREENSHOTS
Ad
Penzai
DESCRIPTION
Penzai, developed by Google DeepMind, is a JAX-based library for representing, visualizing, and manipulating neural network models as functional pytree data structures. It is designed to make machine learning research more interpretable and interactive, particularly for tasks like model surgery, ablation studies, architecture debugging, and interpretability research. Unlike conventional neural network libraries, Penzai exposes the full internal structure of models, enabling fine-grained inspection and modification after training. Its modular design includes tools for tree manipulation, named axes, and declarative neural network construction. The library integrates tightly with Treescope, an advanced pretty-printer for visualizing deeply nested JAX pytrees and NDArray structures. Penzai’s penzai.nn module provides a compositional, combinator-based API for building neural networks.
Features
- Builds and represents models as readable JAX pytrees for full structural transparency
- Provides Treescope for interactive visualization of complex model hierarchies
- Includes pytree manipulation utilities for selective editing and rewrites (pz.select)
- Supports named axes for flexible vectorization and clear computation graphs (pz.nx)
- Combinator-based neural network library (pz.nn) with state and parameter sharing
- Modular transformer implementations for Gemma, LLaMA, Mistral, and GPT-NeoX architectures
Programming Language
Python
Categories
This is an application that can also be fetched from https://sourceforge.net/projects/penzai.mirror/. It has been hosted in OnWorks in order to be run online in an easiest way from one of our free Operative Systems.
