Skip to content

Add Bidirectional module#2

Open
cgarciae wants to merge 66 commits intornn-v2from
add-bidirectional
Open

Add Bidirectional module#2
cgarciae wants to merge 66 commits intornn-v2from
add-bidirectional

Conversation

@cgarciae
Copy link
Copy Markdown
Owner

@cgarciae cgarciae commented Feb 23, 2023

TEST

@cgarciae cgarciae force-pushed the rnn-v2 branch 3 times, most recently from 1d62644 to 83cae20 Compare February 23, 2023 16:05
@cgarciae cgarciae force-pushed the add-bidirectional branch from c0314ad to da12439 Compare March 1, 2023 16:10
IvyZX and others added 27 commits March 9, 2023 15:10
@IvyZX
Allow the checkpoint directory to be nonexistent before saving.
e7b8300 
PiperOrigin-RevId: 515453724
@IvyZX
release 0.6.7
1b11334
Merge pull request google#2944 from IvyZX:push
323ebc6 
PiperOrigin-RevId: 515489629
@IvyZX
bump up head version number
1418373
@chiamp
fixed ScopeParamShapeError message
99ffea0
@8bitmp3
Add Flax Google Research examples
370b1d2
Merge pull request google#2934 from 8bitmp3:google-flax-examples
bdb3f00 
PiperOrigin-RevId: 515880235
Merge pull request google#2946 from IvyZX:push
d1e420a 
PiperOrigin-RevId: 515913880
@cgarciae
permit passing mesh or rules to spmd primitives
97668a9
@cgarciae
add rules and mesh docstrings
d053913
@cgarciae
ignore FutureWarning from jax
d146765
Merge pull request google#2931 from google:fix-tabulate
ccf48a6 
PiperOrigin-RevId: 516388684
@cgarciae
fix mypy errors
5b6579d
Merge pull request google#2939 from google:improve_with_logical_const…
351248c 
…raint

PiperOrigin-RevId: 516678728
@cpgaffney1
Eliminate orbax.checkpoint's dependency on Flax by relying on new jax…
f6d81d4 
….tree_util functions.

PiperOrigin-RevId: 516690676
@saxenasaurabh
Filter out orbax temporary checkpoints from the list of all checkpoints.
6b4c107 
PiperOrigin-RevId: 516697344
@IvyZX
Remove process_index() check in flax/examples, to match Orbax API.
594b452 
PiperOrigin-RevId: 516880068
Merge pull request google#2947 from chiamp:scopeparamshapeerror
94d50f6 
PiperOrigin-RevId: 516954351
@andsteing
Removes wmt/configs/gpu_mixed_precision.py
fec4110 
PiperOrigin-RevId: 517034404
@andsteing
Adds metrics function that describes main metrics of the experiment.
4efd541 
PiperOrigin-RevId: 517040365
@hawkinsp
Replaces references to jax.numpy.DeviceArray with jax.Array.
dacf003 
PiperOrigin-RevId: 517192725
@IvyZX
Fix checkpointing docs for Orbax 0.1.4
a6b86fb
Merge pull request google#2962 from IvyZX:orbax-upgrade-guide
dd14e01 
PiperOrigin-RevId: 517270595
@IvyZX
Migrate Flax away from register_keypaths
d129b66 
PiperOrigin-RevId: 517307374
@IvyZX
Roll back the Orbax flag flip.
751f26a 
PiperOrigin-RevId: 517529153
@yechengxi
set axis='batch'
1c6cb6b 
Without this the accuracy is 71.6 on 4x4.
Adding this the accuracy is 76.5 on 4x4.
Merge pull request google#2967 from yechengxi:patch-2
23340bb 
PiperOrigin-RevId: 518237421
@cgarciae cgarciae force-pushed the add-bidirectional branch from 984ec14 to 74e00f2 Compare March 21, 2023 22:12
@cgarciae cgarciae force-pushed the add-bidirectional branch from 74e00f2 to 6412c6c Compare March 22, 2023 00:13
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Reviewers

No reviews

Assignees

No one assigned

Labels

None yet

Projects

None yet

Milestone

No milestone

Development

Successfully merging this pull request may close these issues.

10 participants

@cgarciae @IvyZX @chiamp @8bitmp3 @yashk2810 @andsteing @hawkinsp @cpgaffney1 @saxenasaurabh @yechengxi