Skip to content

cuPyLMA Release 0.1

Latest
Latest

Choose a tag to compare

View all tags
@wang-xianghao wang-xianghao released this 31 Jul 01:34
· 15 commits to release-0.1 since this release
v0.1
63fe6e7
This commit was created on GitHub.com and signed with GitHub’s verified signature.
GPG key ID: B5690EEEBB952194
Verified
Learn about vigilant mode.

This is a beta release of cuPyLMA.

Changelog

  • First release.

Known issues and Further Work

  • The multi-GPU acceleration is restricted by kernel calls' overheads: we will explore CUDA graph to minimize the overheads.
  • The optimizer does not inherit torch.optim.Optimizer which brings extra work on migrating the existing code: we will reconstruct our optimizer to make it follow PyTorch optimizer's interface.
Assets 3
Loading