Spaces:

flax-community
/

dalle-mini

Running

App Files Files Community

dalle-mini / tools /train /distributed_shampoo.py

Commit History

feat: update distributed_shampoo

b90198c

boris commited on Jan 13, 2022

feat: add shampoo optimizer

0b87452

boris commited on Jan 8, 2022