vit-vqgan

JAX implementation ViT-VQGAN

MIT License

Stars
77
View Code on GitHub View on X
Ecosystems: Python

Statistics for this project are still being loaded, please check back later.

Related Projects
quax

Multiple dispatch over abstract array types in JAX.

20 Jun 2023 101
lineax

Linear solvers in JAX and Equinox. https://docs.kidger.site/lineax

09 May 2023 345
mlp-gpt-jax

A GPT, made only of MLPs, in Jax

21 May 2021 55
rl-finapp

This is JAX implementation of Book of Foundations of Reinforcement Learning with Applications in ...

16 Jan 2024 1
self-attention-cv

Implementation of various self-attention mechanisms focused on computer vision. Ongoing repository.

31 Jan 2021 1,171
Vim

[ICML 2024] Vision Mamba: Efficient Visual Representation Learning with Bidirectional State Space...

15 Jan 2024 2,864
vqgan-jax

JAX implementation of VQGAN

28 Jun 2021 90
flash-attention-jax

Implementation of Flash Attention in Jax

12 Jul 2022 189
lpips-j

Minimal JAX/Flax port of `lpips` supporting `vgg16`, with pre-trained weights stored in the 🤗 Hug...

08 Jul 2022 14
PaLM-rlhf-pytorch

Implementation of RLHF (Reinforcement Learning with Human Feedback) on top of the PaLM architectu...

09 Dec 2022 7,595
vrp-attention-jax

Reimplementation of the paper "Attention, Learn to Solve Routing Problems!" in jax/flax.

03 Oct 2021 7
simple-diffusion

An implementation of simple diffusion in PyTorch (and JAX)

27 Jan 2023 18
PaLM-jax

Implementation of the specific Transformer architecture from PaLM - Scaling Language Modeling wit...

08 Apr 2022 184
equinox

Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/

29 Jul 2021 1,846
lightweight-gan

Implementation of 'lightweight' GAN, proposed in ICLR 2021, in Pytorch. High resolution image gen...

11 Nov 2020 1,623