LOMO

What's Changed

• fix ds_shape error
• fix linked images in PyPI page by @Borda in https://github.com/OpenLMLab/LOMO/pull/76
• FIX: avoid math errors for edge cases by @younesbelkada in https://github.com/OpenLMLab/LOMO/pull/78

New Contributors

• @Borda made their first contribution in https://github.com/OpenLMLab/LOMO/pull/76
• @younesbelkada made their first contribution in https://...

Uploading an installable package to PyPI.

pip install lomo-optim

Then, you can import Lomo or AdaLomo

from lomo_optim import Lomo, AdaLomo

• Add AdaLomo, a more powerful and equally efficient optimizer. https://arxiv.org/abs/2310.10195.
• The code is easier to use based on collie (dev-branch currently).