Skip to content

Pull requests: google-deepmind/penzai

New pull request New
5 Open 88 Closed
5 Open 88 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

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! Updated in the last three days: updated:>2025-10-16.