optax-galore

Python ★ 5 updated 1y ago

Adds GaLore style projection wrappers to optax optimizers

No plain-English explanation yet — one is being written right now. Check back in a minute.