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.
请按照以下说明运行此应用程序：
- 1. 在您的 PC 中下载此应用程序。
- 2. 在我们的文件管理器 https://www.onworks.net/myfiles.php?username=XXXXX 中输入您想要的用户名。
- 3. 在这样的文件管理器中上传这个应用程序。
- 4. 从此网站启动OnWorks Linux online 或Windows online emulator 或MACOS online emulator。
- 5. 从您刚刚启动的 OnWorks Linux 操作系统，使用您想要的用户名转到我们的文件管理器 https://www.onworks.net/myfiles.php?username=XXXXX。
- 6. 下载应用程序，安装并运行。
SCREENSHOTS
Ad
Penzai
商品描述
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.
功能
- 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
程式语言
Python
分类
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.
