flax
[nnx] Add complex test with scan + batchnorm + dropout
#3567

Merged

[nnx] Add complex test with scan + batchnorm + dropout #3567

copybara-service merged 1 commit into main from nnx-add-dropout-test

nnx: add complex test with scan + batchnorm + dropout

df60f611

cgarciae requested a review from

chiamp 2 years ago

chiamp approved these changes on 2023-12-19

chiamp added pull ready

copybara-service merged 3e56f8e4 into main 2 years ago

copybara-service deleted the nnx-add-dropout-test branch 2 years ago

Reviewers

chiamp

Assignees

No one assigned

Labels

pull ready

Milestone

No milestone