Skip to content

Pull requests: google-deepmind/penzai

New pull request New
10 Open 90 Closed
10 Open 90 Closed
Author
Filter by author
Loading
Label
Filter by label
Loading
Use alt + click/return to exclude labels
or ⇧ + click/return for logical OR
Projects
Filter by project
Loading
Milestones
Filter by milestone
Loading
Reviews
Filter by reviews
No reviews Review required Approved review Changes requested
Assignee
Filter by who’s assigned
Assigned to nobody Loading
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

Pull requests list

Support Optax extra arguments in StatefulTrainer
#143 opened Aug 16, 2026 by sylvesterkaczmarek Loading…
1
Allow hashable axis names in shape checks
#142 opened Aug 15, 2026 by sylvesterkaczmarek Loading…
1
Allow complex values in Linear layers
#141 opened Aug 15, 2026 by sylvesterkaczmarek Loading…
1
Add Gemma-4 support
#139 opened Aug 10, 2026 by JEM-Mosig Contributor • Draft
feat: Add Geometric Sparse Attention (AETHER)
#134 opened Jan 20, 2026 by teerthsharma Loading…
1
Add py.typed marker.
#130 opened Oct 17, 2025 by Jammf Loading…
Fix dtype type annotations to use jax.typing.DTypeLike.
#129 opened Oct 16, 2025 by Jammf Loading…
Use jax.tree_util.treedef_is_leaf for readability
#128 opened Aug 19, 2025 by findmyway Loading…
Feature Request: Add convolution layers
#120 opened May 31, 2025 by AntoinePlumerault • Draft
41
Experiment with using out_axes=-1 inside nmap's vmap
#99 opened Nov 25, 2024 by shoyer • Draft
ProTip! Adding no:label will show everything without a label.