flax.linen.apply takes state as the first param, followed by the input (#12510)
flax.linen.apply takes state as the first param, followed by the input (#12510)


huggingface/transformers,huggingface/transformers,huggingface/transformers,huggingface/transformers