penzai

A JAX research toolkit for building, editing, and visualizing neural networks.

APACHE-2.0 License

Downloads
2.3K
Stars
1.6K
View Code on GitHub Visit Website
Ecosystems: Python

Issue Statistics

Past Year

All Time

Total Pull Requests
69
69
Merged Pull Requests
52
52
Total Issues
15
15
Time to Close Issues
15 days
15 days
Package Rankings
Top 36.21% on Pypi.org
Related Projects
FastChat

An open platform for training, serving, and evaluating large language models. Release repo for Vi...

19 Mar 2023 36,628
viewer

ML models and internal tensors 3D visualizer

17 Feb 2021 1,106
equinox

Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/

29 Jul 2021 1,846
VAR

[GPT beats diffusion🔥] [scaling laws in visual generation📈] Official impl. of "Visual Autoregress...

01 Apr 2024 2,568
dm-haiku

JAX-based neural network library

18 Feb 2020 2,816
PixArt-alpha

PixArt-α: Fast Training of Diffusion Transformer for Photorealistic Text-to-Image Synthesis

12 Oct 2023 2,138
the-incredible-pytorch

The Incredible PyTorch: a curated list of tutorials, papers, projects, communities and more relat...

11 Feb 2017 11,389
minGPT-flax

GPT implementation in Flax

30 Sep 2021 16
torchview

torchview: visualize pytorch models

08 Nov 2022 803
ninjax

General Modules for JAX

23 May 2022 57
simple-onnx-processing-tools

A set of simple tools for splitting, merging, OP deletion, size compression, rewriting attributes...

11 Apr 2022 268
SPADE

Semantic Image Synthesis with SPADE

14 Mar 2019 7,528
jraph

A Graph Neural Network Library in Jax

23 Nov 2020 1,327
pytorch-widedeep

A flexible package for multimodal-deep-learning to combine tabular data with text and images usin...

21 Oct 2017 1,243
lm-evaluation-harness

A framework for few-shot evaluation of language models.

28 Aug 2020 6,569