patil-suraj/vqgan-jax

JAX implementation of VQGAN

91

stars

18

commits

Python

primary language

Jul 9, 2022

updated

README

vqgan-jax

[WIP] JAX implementation of taming-transformers VQGAN.

(Note: For now this only consits of the VQModel for inference. The discriminatior, transformer model and training scripts will be added later.)

Use this to see how to use this model to encode and reconstruct images.

Contributors

10 commits

6 commits

2 commits

Similar repos

borisdayma/dalle-mini

DALL·E Mini - Generate images from a text prompt

14,732

shared contributors

dalle-mini/vqgan_imagenet_f16_16384

42

dfdx/gendo

Generative models

2

alif-munim/echo-vqgan

ViT-VQGAN for echocardiogram quality enhancement.

1

KohakuBlueleaf/HakuLatent

A comprehensive codebase for training and finetuning Image <> Latent models.

49

sl33pyC01E/mini-vae

0

SmilingWolf/JAX-CV

Repo for my JAX CV experiments. Mostly revolving around the Danbooru20xx dataset

49

google-research/magvit

Official JAX implementation of MAGVIT: Masked Generative Video Transformer

1,001

Languages

Python

100.0%

patil-suraj/vqgan-jax

JAX implementation of VQGAN

91

stars

18

commits

Python

primary language

Jul 9, 2022

updated

README

vqgan-jax

[WIP] JAX implementation of taming-transformers VQGAN.

(Note: For now this only consits of the VQModel for inference. The discriminatior, transformer model and training scripts will be added later.)

Use this to see how to use this model to encode and reconstruct images.

Similar repos

borisdayma/dalle-mini

DALL·E Mini - Generate images from a text prompt

14,732

shared contributors

dalle-mini/vqgan_imagenet_f16_16384

42

dfdx/gendo

Generative models

2

alif-munim/echo-vqgan

ViT-VQGAN for echocardiogram quality enhancement.

1

KohakuBlueleaf/HakuLatent

A comprehensive codebase for training and finetuning Image <> Latent models.

49

sl33pyC01E/mini-vae

0

SmilingWolf/JAX-CV

Repo for my JAX CV experiments. Mostly revolving around the Danbooru20xx dataset

49

google-research/magvit

Official JAX implementation of MAGVIT: Masked Generative Video Transformer

1,001

Contributors

10 commits

6 commits

2 commits

Languages

Python

100.0%