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

Merged

copybara-service[bot] merged 1 commit intomainfrom test_675988367

Sep 23, 2024

+13-1

Commits on Sep 23, 2024

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

nimrod-gileadi
authored and
Flax Authors
committed