flax

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