Skip to content

Pull requests: google-deepmind/dm-haiku

New pull request New
13 Open 584 Closed
13 Open 584 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

Avoid usage of deprecated jax.core APIs.
#879 opened Apr 15, 2026 by copybara-service Bot Loading…
Add runtime deprecation warnings to one_hot and multinomial
#865 opened Jan 30, 2026 by t-suraj21 Loading…
1
Upgrade GitHub Actions for Node 24 compatibility
#863 opened Jan 25, 2026 by salmanmkc Loading…
Improve beginner explanations in Haiku basics notebook
#861 opened Jan 19, 2026 by MukarramBambot Loading…
1
fix(batch_norm): allow string cross_replica_axis for vmap support
#859 opened Jan 8, 2026 by aditya161205 Loading…
2
1
[pmap] Remove jax_pmap_shmap_merge configuration.
#850 opened Dec 3, 2025 by copybara-service Bot Loading…
Add Feedback Alignment implementation
#848 opened Nov 1, 2025 by hwilner Loading…
1
1
Disable running tests on 3.1{0,1}.
#839 opened Aug 4, 2025 by copybara-service Bot Loading…
Fix incorrect type specification for Haiku partition function.
#826 opened Jun 11, 2025 by copybara-service Bot Loading…
1
fixing avg_pool wrong shape issue
#824 opened Jun 1, 2025 by Vasist10 Loading…
1
Missing MaxPool2d
#815 opened Apr 5, 2025 by DEBADAS001KERNEL Loading…
Internal change.
#547 opened Oct 25, 2022 by copybara-service Bot Loading…
1
[JAX] Add an option to control the out-of-bounds indexing behavior of Haiku embedding layers.
#493 opened Aug 5, 2022 by copybara-service Bot Loading…
ProTip! no:milestone will show everything without a milestone.