Function flatA

flatA(lora: TrainableLora): MlxArray[]
A leaves in stable (sorted-by-modulePath) order — the order used for value_and_grad argnums and AdamW state.
Parameters
- lora: TrainableLora
Returns MlxArray[]
- Defined in packages/training/src/lora-params.ts:141