Skip to content

Pull requests: google/learned_optimization

New pull request New
42 Open 276 Closed
42 Open 276 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

[pmap] Remove conditional jax.config.jax_pmap_shmap_merge logic as part of last phase of jax.pmap clean-up.
#336 opened Feb 2, 2026 by copybara-service bot Loading…
Stackless yashful
#326 opened Oct 3, 2024 by copybara-service bot Loading…
Update libraries to use JAX's limited (and ill-advised) trace-state-querying APIs rather than depending on JAX's deeper internals, which are about to change.
#324 opened Sep 24, 2024 by copybara-service bot Loading…
No public description
#323 opened Mar 26, 2024 by copybara-service bot Loading…
1
Update README.md
#321 opened Jan 19, 2024 by Sohl-Dickstein Loading…
Lopt conv+ln
#320 opened Jan 4, 2024 by AllanYangZhou Loading…
Fix various type annotations.
#306 opened Oct 11, 2023 by copybara-service bot Loading…
Internal change
#298 opened Jul 14, 2023 by copybara-service bot Loading…
1
Internal change
#297 opened Jul 14, 2023 by copybara-service bot Loading…
1
Reshape images to 2D so imshow can show them
#276 opened Jan 18, 2023 by grofte Loading…
Monkey-patch os.path to support gs:// URIs
#273 opened Dec 19, 2022 by janEbert Loading…
Internal change
#238 opened Aug 19, 2022 by copybara-service bot Loading…
Internal change
#236 opened Aug 10, 2022 by copybara-service bot Loading…
1
Internal change
#235 opened Aug 10, 2022 by copybara-service bot Loading…
1
Internal change
#234 opened Aug 10, 2022 by copybara-service bot Loading…
1
Internal change
#233 opened Aug 10, 2022 by copybara-service bot Loading…
1
Internal change
#227 opened Aug 5, 2022 by copybara-service bot Loading…
1
Internal change
#226 opened Aug 4, 2022 by copybara-service bot Loading…
1
Internal change
#225 opened Aug 4, 2022 by copybara-service bot Loading…
1
Internal change
#224 opened Aug 4, 2022 by copybara-service bot Loading…
1
Internal change
#223 opened Aug 4, 2022 by copybara-service bot Loading…
1
Internal change
#222 opened Aug 4, 2022 by copybara-service bot Loading…
1
Internal change
#221 opened Aug 4, 2022 by copybara-service bot Loading…
1
Internal change
#220 opened Aug 4, 2022 by copybara-service bot Loading…
1
Internal change
#219 opened Aug 4, 2022 by copybara-service bot Loading…
1
ProTip! Follow long discussions with comments:>50.