Skip to content

Pull requests: google-research/vmoe

New pull request New
Clear current search query, filters, and sorts
8 Open 156 Closed
8 Open 156 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
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

Pull requests list

[numpy] Fix users of NumPy APIs that are removed in NumPy 2.0.
#180 by copybara-service bot was merged Jul 25, 2024 Loading…
Internal change.
#179 by copybara-service bot was merged Jul 9, 2024 Loading…
Stop writing msgpack file for new checkpoints and update empty nodes handling so that it no longer depends on this file.
#178 by copybara-service bot was merged Jul 8, 2024 Loading…
Stop writing msgpack file for new checkpoints and update empty nodes handling so that it no longer depends on this file.
#177 by copybara-service bot was merged Jul 3, 2024 Loading…
Fix initialization from Orbax checkpoint.
#174 by copybara-service bot was closed May 28, 2024 Loading…
Replace deprecated jax.tree_* functions with jax.tree.*
#173 by copybara-service bot was merged May 15, 2024 Loading…
Internal changes.
#172 by copybara-service bot was merged Apr 29, 2024 Loading…
Remove write_tree_metadata option, as there is no known reason to disable this.
#171 by copybara-service bot was merged Apr 24, 2024 Loading…
Remove write_tree_metadata option, as there is no known reason to disable this.
#170 by copybara-service bot was closed May 3, 2024 Loading…
Silence some pytype errors.
#169 by copybara-service bot was merged Apr 15, 2024 Loading…
Release config and code necessary to pretrain on LAION-400M.
#168 by copybara-service bot was closed Apr 5, 2024 Loading…
Access thread_resources via jax.interpreters.pxla instead of jax.experimental.maps
#167 by copybara-service bot was merged Mar 24, 2024 Loading…
Report training flops and optimal device time computed by XLA while training.
#166 by copybara-service bot was merged Mar 22, 2024 Loading…
Replace jax.tree_(*) functions with jax.tree_util.tree_\1.
#165 by copybara-service bot was closed Mar 12, 2024 Loading…
Use shard_map instead of xmap
#164 by copybara-service bot was merged Feb 28, 2024 Loading…
Use shard_map instead of xmap
#163 by copybara-service bot was merged Feb 27, 2024 Loading…
Removed unnecessary pjit/xmap hack in ZoomTransformation
#162 by copybara-service bot was closed Feb 27, 2024 Loading…
Release pretraining config on ImageNet-21k, used in the original V-MoE paper.
#161 by copybara-service bot was closed Feb 7, 2024 Loading…
Fix bug by changes in Big Vision.
#158 by copybara-service bot was closed Dec 5, 2023 Loading…
Update EEE paper fine-tuning config after internal updates.
#157 by copybara-service bot was closed Nov 20, 2023 Loading…
Minor internal changes.
#155 by copybara-service bot was merged Dec 1, 2023 Loading…
Add text tokenization preprocessing op.
#154 by copybara-service bot was merged Nov 19, 2023 Loading…
Update make_create_train_state_fn to support multiple model inputs.
#153 by copybara-service bot was closed Nov 19, 2023 Loading…
Remove references to deprecated jax.nn.normalize
#150 by copybara-service bot was closed Nov 9, 2023 Loading…
Write tree metadata when checkpointing.
#149 by copybara-service bot was closed Oct 30, 2023 Loading…
ProTip! Exclude everything labeled bug with -label:bug.