Search results

Gemma 4 Support

Status: Open.

#137 In google-deepmind/penzai;

· JEM-Mosig opened on Apr 13, 2026

[RFC] Geometric Sparse Attention Layer (AETHER)

Status: Open.

#133 In google-deepmind/penzai;

· teerthsharma opened on Jan 20, 2026

BUG: pz.chk.check_structure erroneously requires all named array axes to be strings.

Status: Open.

#132 In google-deepmind/penzai;

· amifalk opened on Jan 18, 2026

Using Complex Numbers in the Linear layer does not work.

Status: Open.

#127 In google-deepmind/penzai;

· beegica opened on Aug 11, 2025

Passing extra arguments to gradient transformations

Status: Open.

#126 In google-deepmind/penzai;

· neverix opened on Aug 1, 2025

LayerStack.from_sublayer_builder temporarily uses about twice the memory

bugSomething isn't workingSomething isn't working

Status: Open.

#122 In google-deepmind/penzai;

· JEM-Mosig opened on Jun 7, 2025

strange import behavior of penzai.nn and pz

Status: Open.

#114 In google-deepmind/penzai;

· hrbigelow opened on Apr 17, 2025

Feature request: Support for quantization

feature-requestNew feature or requestNew feature or request

Status: Open.

#113 In google-deepmind/penzai;

· JEM-Mosig opened on Apr 17, 2025

Initializer for Penzai's Linear layer does not support jax.nn.initializers

Status: Open.

#107 In google-deepmind/penzai;

· Samarendra109 opened on Feb 7, 2025

FR: expose device/sharding argument for NamedArray.wrap

Status: Open.

#80 In google-deepmind/penzai;

· amifalk opened on Aug 8, 2024

FR: batched mapping for named_axes.nmap

feature-requestNew feature or requestNew feature or request

Status: Open.

#71 In google-deepmind/penzai;

· amifalk opened on Jul 26, 2024