flax

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

#4208

Merged