Skip to content

Pull requests: google/init2winit

New pull request New
69 Open 768 Closed
69 Open 768 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

Update jax.host_id() and jax.host_count() to jax.process_index() and jax.process_count(), respectively.
#839 opened Jan 30, 2026 by copybara-service bot Loading…
Switch WMT default attention to JAX built-in attention, switch default launch script to be on one cell for both compute & storage
#836 opened Jan 27, 2026 by copybara-service bot Loading…
1
Migrate from flax checkpoint to Orbax CheckpointManager.
#835 opened Jan 13, 2026 by copybara-service bot Loading…
Use jnp for metrics to make arrays addressable in multi-task setting.
#834 opened Jan 8, 2026 by copybara-service bot Loading…
internal change
#827 opened Dec 10, 2025 by copybara-service bot Loading…
temporary line, public changes are in parent
#814 opened Oct 20, 2025 by copybara-service bot Loading…
internal change
#810 opened Oct 9, 2025 by copybara-service bot Loading…
internal change
#807 opened Oct 7, 2025 by copybara-service bot Loading…
internal change
#804 opened Oct 1, 2025 by copybara-service bot Loading…
Clean up unused args in base_trainer. These were introduced by the quantization trainer which was removed over a year ago.
#798 opened Sep 5, 2025 by copybara-service bot Loading…
1
extending bubbles to general tensors
#787 opened Jul 10, 2025 by copybara-service bot Loading…
1
Enable local launches using debug configs in launch_multitrain_xm_v2.py (in particular, wmt17_de_en_transformer workload can now be debugged locally)
#786 opened Jul 4, 2025 by copybara-service bot Loading…
Internal
#766 opened May 15, 2025 by copybara-service bot Loading…
Internal
#760 opened May 5, 2025 by copybara-service bot Loading…
Internal change.
#740 opened Apr 9, 2025 by copybara-service bot Loading…
1
Internal change
#737 opened Apr 1, 2025 by copybara-service bot Loading…
Internal change.
#730 opened Mar 28, 2025 by copybara-service bot Loading…
internal change
#716 opened Mar 1, 2025 by copybara-service bot Loading…
1
Internal
#705 opened Jan 30, 2025 by copybara-service bot Loading…
Update references to JAX's GitHub repo
#690 opened Dec 5, 2024 by copybara-service bot Loading…
Bump tensorflow from 2.5.0 to 2.12.1 in the pip group dependencies Pull requests that update a dependency file
#677 opened Jul 30, 2024 by dependabot bot Loading…
Stop writing msgpack file for new checkpoints and update empty nodes handling so that it no longer depends on this file.
#673 opened Jun 27, 2024 by copybara-service bot Loading…
Adding schedule free with adam and nadam
#672 opened Jun 12, 2024 by copybara-service bot Loading…
1
Internal.
#670 opened May 14, 2024 by copybara-service bot Loading…
Proposal to make 'eval_use_ema' a hyper-parameter visible to optimizers.
#669 opened May 14, 2024 by copybara-service bot Loading…
ProTip! Type g p on any issue or pull request to go back to the pull request listing page.