axlearn
axlearn copied to clipboard
Update lora's param_partition_spec.
@JianyuWangV You may need to rebase your changes to get CI to run. Thanks!
Do we still need this PR?
Do we still need this PR?
Yes, thanks for the reminder! Let me rebase the PR and test it.