Alternatives and similar repositories for CLIP_JAX

Users that are interested in CLIP_JAX are comparing it to the libraries listed below

• crowsonkb / cloob-training
  CLOOB training (JAX) and inference (JAX and PyTorch)
  ☆72Updated 3 years ago
• matthias-wright / flaxmodels
  Pretrained deep learning models for Jax/Flax: StyleGAN2, GPT2, VGG, ResNet, etc.
  ☆254Updated 3 months ago
• crowsonkb / v-diffusion-jax
  v objective diffusion inference code for JAX.
  ☆214Updated 3 years ago
• crowsonkb / consistency-models
  A JAX implementation of the continuous time formulation of Consistency Models
  ☆85Updated 2 years ago
• lucidrains / ponder-transformer
  Implementation of a Transformer that Ponders, using the scheme from the PonderNet paper
  ☆81Updated 3 years ago
• JD-P / cloob-latent-diffusion
  CLOOB Conditioned Latent Diffusion training and inference code
  ☆113Updated 3 years ago
• patil-suraj / stable-diffusion-jax
  ☆89Updated 2 years ago
• crowsonkb / esgd
  ESGD-M is a stochastic non-convex second order optimizer, suitable for training deep learning models, for PyTorch.
  ☆56Updated 2 years ago
• lucidrains / panoptic-transformer
  Another attempt at a long-context / efficient transformer by me
  ☆38Updated 3 years ago
• Jack000 / DALLE-pytorch
  Implementation / replication of DALL-E, OpenAI's Text to Image Transformer, in Pytorch
  ☆89Updated 3 years ago
• iejMac / video2numpy
  Optimized library for large-scale extraction of frames and audio from video.
  ☆204Updated last year
• NVlabs / LACE
  A new play-and-plug method of controlling an existing generative model with conditioning attributes and their compositions.
  ☆73Updated 3 years ago
• tgisaturday / dalle-lightning
  Refactoring dalle-pytorch and taming-transformers for TPU VM
  ☆60Updated 3 years ago
• patil-suraj / vit-vqgan
  JAX implementation ViT-VQGAN
  ☆83Updated 2 years ago
• lucidrains / PaLM-jax
  Implementation of the specific Transformer architecture from PaLM - Scaling Language Modeling with Pathways - in Jax (Equinox framework)
  ☆187Updated 3 years ago
• lucidrains / hourglass-transformer-pytorch
  Implementation of Hourglass Transformer, in Pytorch, from Google and OpenAI
  ☆91Updated 3 years ago
• borisdayma / clip-jax
  Train vision models using JAX and 🤗 transformers
  ☆98Updated 2 months ago
• patil-suraj / vqgan-jax
  JAX implementation of VQGAN
  ☆93Updated 2 years ago
• AranKomat / Diff-DALLE
  ☆64Updated 3 years ago
• shyamsn97 / hyper-nn
  Easy Hypernetworks in Pytorch and Jax
  ☆101Updated 2 years ago
• lucidrains / mlp-gpt-jax
  A GPT, made only of MLPs, in Jax
  ☆58Updated 4 years ago
• cfoster0 / simple-diffusion-model
  Pedagogical codebase for a simplified score-based generative model design, with training loop
  ☆39Updated 3 years ago
• HomebrewML / revlib
  Simple and efficient RevNet-Library for PyTorch with XLA and DeepSpeed support and parameter offload
  ☆127Updated 2 years ago
• lucidrains / flexible-diffusion-modeling-videos-pytorch
  Implementation of the video diffusion model and training scheme presented in the paper, Flexible Diffusion Modeling of Long Videos, in Py…
  ☆85Updated 3 years ago
• lucidrains / token-shift-gpt
  Implementation of Token Shift GPT - An autoregressive model that solely relies on shifting the sequence space for mixing
  ☆50Updated 3 years ago
• lucidrains / g-mlp-gpt
  GPT, but made only out of MLPs
  ☆89Updated 4 years ago
• DarshanDeshpande / jax-models
  Unofficial JAX implementations of deep learning research papers
  ☆156Updated 2 years ago
• HomebrewML / Olmax
  HomebrewNLP in JAX flavour for maintable TPU-Training
  ☆50Updated last year
• ml-jku / cloob
  ☆160Updated 3 years ago
• yiyixuxu / denoising-diffusion-flax
  Implementing the Denoising Diffusion Probabilistic Model in Flax
  ☆149Updated 2 years ago