search

patil-suraj / vit-vqgan

JAX implementation ViT-VQGAN
MIT License
77 stars 11 forks source link

feat: update shampoo #25

Closed borisdayma closed 2 years ago

borisdayma commented 2 years ago

Update shampoo to use the latest and greatest version!