vqgan-jax

JAX implementation of VQGAN

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

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

Badges
Extracted from project README
Colab
Related Projects
transformers

🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.

29 Oct 2018 132,459
attax

Attax: adversarial attacks using JAX

27 Oct 2019 3
quax

Multiple dispatch over abstract array types in JAX.

20 Jun 2023 101
lpips-j

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

08 Jul 2022 14
vit-vqgan

JAX implementation ViT-VQGAN

22 Jun 2022 77
VILA

VILA - a multi-image visual language model with training, inference and evaluation recipe, deploy...

23 Feb 2024 1,061
TransformerEngine

A library for accelerating Transformer models on NVIDIA GPUs, including using 8-bit floating poin...

20 Sep 2022 1,482
n-grammer-flax

Implementation of N-Grammer in Flax

04 Aug 2022 14
mm-cot

Official implementation for "Multimodal Chain-of-Thought Reasoning in Language Models" (stay tune...

02 Feb 2023 3,760
minGPT-flax

GPT implementation in Flax

30 Sep 2021 16
Lightweighting_Cookbook

This project attempts to build neural network training and lightweighting cookbook including thre...

25 May 2022 18
equinox

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

29 Jul 2021 1,846
fidjax

Frechet Inception Distance in JAX

15 Jul 2024 1
train-text-to-image-tpu-tutorial

20 Jan 2023 19
vrp-attention-jax

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

03 Oct 2021 7