flax
Forward all arguments when using nnx.transforms.deprecated.scan as a decorator.
#4208

Merged

Forward all arguments when using nnx.transforms.deprecated.scan as a decorator. #4208

copybara-service merged 1 commit into main from test_675988367

copybara-service force pushed 1 year ago

Forward all arguments when using nnx.transforms.deprecated.scan as a …

2dc30a8a

copybara-service force pushed to 2dc30a8a 1 year ago

copybara-service merged 2dc30a8a into main 1 year ago

copybara-service deleted the test_675988367 branch 1 year ago

Reviewers

No reviews

Assignees

No one assigned

Labels

None yet

Milestone

No milestone