Skip to content

Conversation

@zitongzhan
Copy link
Contributor

@zitongzhan zitongzhan commented Mar 1, 2023

#149

  • Made the function a torch.nn.Module
  • Batched operation. However, does not allow arbitrary batch shape (...) because of the limitations of svd and vmap functions
  • Refactored all naming
  • Added docstring

Diff with original implementation. Naming are changed to all lower case but correspondence are still identifiable.

@zitongzhan zitongzhan requested a review from wang-chen March 1, 2023 01:50
Copy link
Member

@wang-chen wang-chen left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Following our meeting, better to give an API design first: showing how we want the user to use this module. Just like the example we provided in LM optimizer.

@zitongzhan zitongzhan requested a review from wang-chen April 4, 2023 20:01
@zitongzhan
Copy link
Contributor Author

zitongzhan commented Apr 9, 2023

Summary

@zitongzhan zitongzhan requested a review from jingnanshi April 10, 2023 02:58
Copy link
Contributor

@jingnanshi jingnanshi left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Great job! I just left a few minor comments.

@zitongzhan zitongzhan merged commit 5894281 into main Apr 12, 2023
@zitongzhan zitongzhan deleted the feature/epnp branch April 12, 2023 01:29
@wang-chen wang-chen mentioned this pull request May 6, 2023
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

4 participants