flax
Add support for layer-specific rope scale factors.
#4672
Merged

Loading