flax
Add support for initializing variables_carry in nn.scan
#3228

Open

Add support for initializing variables_carry in nn.scan #3228

cgarciae wants to merge 3 commits into main from improve-scan-2

add support for initializing carry variables in scan

b12efcc0

cgarciae requested a review from

IvyZX 2 years ago

IvyZX commented on 2023-07-27

tabulate with eval_shape on scan

5a0aae73

IvyZX approved these changes on 2023-08-03

jheek approved these changes on 2023-08-04

remove tree_map_upto_left

748677c4

IvyZX added pull ready

Reviewers

IvyZX

jheek

Assignees

No one assigned

Labels

pull ready

Milestone

No milestone