transformers

Replace default split function with jnp.split() in flax models

#37001

Merged