Support scan layers - Githubissues

patil-suraj / vit-vqgan

JAX implementation ViT-VQGAN

MIT License

77 stars 11 forks source link

Open borisdayma opened 2 years ago

borisdayma commented 2 years ago

It could make compilation faster but e logic is complex (especially for shampoo)