transformers
4cc65e99 - Replace default split function with jnp.split() in flax models (#37001)

Commit

267 days ago

Replace default split function with jnp.split() in flax models (#37001) Replace split with jnp's split function for flax models (#36854)

References

Author

premmurugan229

Parents