JAX/Flax implementation of "Relaxed Recursive Transformers: Effective Parameter Sharing with Layer-wise LoRA" (Bae et al., ICLR 2025, Google DeepMind)
-
Updated
Feb 12, 2026 - Python
#
jax-flax-transformers-lora
Here is 1 public repository matching this topic...
Improve this page
Add a description, image, and links to the jax-flax-transformers-lora topic page so that developers can more easily learn about it.
Add this topic to your repo
To associate your repository with the jax-flax-transformers-lora topic, visit your repo's landing page and select "manage topics."