google-deepmind/penzai
View on GitHub ↗ ★ 1,899 Python archived
A JAX research toolkit for building, editing, and visualizing neural networks.
5 mentions on Hacker News by 5 people: 2 stories, 3 comments · first 2024-04-19 · latest 2025-09-10
20082026
-
…in the JAX/XLA world). I've hit this many times and got batch variance explained to me before: https://github.com/google-deepmind/penzai/issues/82 and https://github.com/jax-ml/jax/issues/20047#issuecomment-1975...
comment by daralthus · 2025-09-10 · in Defeating Nondeterminism in LLM Inference -
…https://github.com/ofnote/tsalib https://github.com/google-deepmind/tensor_annotations https://github.com/google-deepmind/penzai https://github.com/tensorflow/mesh/ https://github.com/facebookresearch/torchdim https://xarray.dev/…
comment by albertzeyer · 2025-05-16 · in I don't like NumPy -
For object inspection in Jupyter if doing more numpy type things, check out penzai: https://github.com/google-deepmind/penzai
comment by randomfool · 2024-07-26 · in Show HN: Wat – Deep inspection of Python objects -
▲ 261
Penzai: JAX research toolkit for building, editing, and visualizing neural netsstory by mccoyb · 2024-04-21 · 51 comments
-
▲ 3
Penzai: JAX Research Toolkit for Neural Networksstory by ubj · 2024-04-19 · 0 comments