PR #648 Add optimized LSTM cell.

Add optimized LSTM cell. #648

copybara-service merged 14 commits into master from optimized_lstm

Add optimized LSTM cell.

86c80c6f

google-cla added cla: yes

Remove unnessary import from nn_test.

eb12ad84

bastings requested a review from

marcvanzee 5 years ago

avital assigned

bastings 5 years ago

avital added Priority: P2 - no schedule

avital added Status: in progress

avital added Type: enhancement

Add OptimizedLSTMCell Linen version.

5135d952

Merge with main branch

deda93be

Fix initializer for DummyDense.

3415c204

Fix initializers.

36fae9a8

Fix return types.

1f35eb4e

Fix type.

094e2d9d

bastings marked this pull request as draft 5 years ago

bastings marked this pull request as ready for review 5 years ago

marcvanzee requested changes on 2020-12-07

avital commented on 2020-12-08

Apply suggestions from code review

ed67ff13

Rename DummyDense and make _concat_dense more readable.

895aa1dc

bastings requested a review from

avital 5 years ago

bastings requested a review from

marcvanzee 5 years ago

marcvanzee approved these changes on 2020-12-09

Simplify _concat_dense using jnp.dot.

f5931862

Fix ConcretizationTypeError in jnp.split.

331bdc2c

Merge remote-tracking branch 'origin/master' into optimized_lstm

24614827

bastings removed review request from

avital 5 years ago

Remove trailing whitespace.

e7a8d335

copybara-service merged 692a62c5 into master 5 years ago

copybara-service deleted the optimized_lstm branch 5 years ago

Reviewers

marcvanzee

avital

Assignees

bastings

Labels

cla: yes Priority: P2 - no schedule

Milestone

No milestone

flax Add optimized LSTM cell. #648 Merged

Add optimized LSTM cell. #648

flax
Add optimized LSTM cell.
#648

Merged