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)
  ☆70Updated 2 years ago
• crowsonkb / v-diffusion-jax
  v objective diffusion inference code for JAX.
  ☆213Updated 2 years ago
• patil-suraj / stable-diffusion-jax
  ☆88Updated 2 years ago
• Jack000 / DALLE-pytorch
  Implementation / replication of DALL-E, OpenAI's Text to Image Transformer, in Pytorch
  ☆89Updated 3 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
• borisdayma / clip-jax
  Train vision models using JAX and 🤗 transformers
  ☆96Updated last month
• patil-suraj / vit-vqgan
  JAX implementation ViT-VQGAN
  ☆82Updated 2 years ago
• lucidrains / ponder-transformer
  Implementation of a Transformer that Ponders, using the scheme from the PonderNet paper
  ☆80Updated 3 years ago
• rom1504 / python-template
  Simple python template
  ☆40Updated 10 months ago
• rom1504 / gpu-tester
  gpu tester detects broken and slow gpus in a cluster
  ☆68Updated 2 years ago
• matthias-wright / flaxmodels
  Pretrained deep learning models for Jax/Flax: StyleGAN2, GPT2, VGG, ResNet, etc.
  ☆249Updated last year
• multimodalart / latent-diffusion-notebook
  ☆57Updated 2 years ago
• JD-P / cloob-latent-diffusion
  CLOOB Conditioned Latent Diffusion training and inference code
  ☆112Updated 2 years ago
• AranKomat / Diff-DALLE
  ☆64Updated 3 years ago
• lucidrains / PaLM-jax
  Implementation of the specific Transformer architecture from PaLM - Scaling Language Modeling with Pathways - in Jax (Equinox framework)
  ☆187Updated 2 years ago
• crowsonkb / guided-diffusion
  ☆151Updated last year
• davisyoshida / lorax
  LoRA for arbitrary JAX models and functions
  ☆135Updated last year
• crowsonkb / LDLM
  Latent Diffusion Language Models
  ☆68Updated last year
• crowsonkb / consistency-models
  A JAX implementation of the continuous time formulation of Consistency Models
  ☆84Updated last year
• tgisaturday / dalle-lightning
  Refactoring dalle-pytorch and taming-transformers for TPU VM
  ☆60Updated 3 years ago
• shyamsn97 / hyper-nn
  Easy Hypernetworks in Pytorch and Jax
  ☆98Updated 2 years ago
• kingoflolz / cc_img_dl
  ☆27Updated 4 years ago
• ClashLuke / tpucare
  Automatically take good care of your preemptible TPUs
  ☆36Updated last year
• lucidrains / mlp-gpt-jax
  A GPT, made only of MLPs, in Jax
  ☆57Updated 3 years ago
• lucidrains / panoptic-transformer
  Another attempt at a long-context / efficient transformer by me
  ☆37Updated 2 years ago
• sholtodouglas / scalingExperiments
  ☆58Updated 3 years ago
• pbaylies / Augmented_CLIP
  Training simple models to predict CLIP image embeddings from text embeddings, and vice versa.
  ☆60Updated 2 years ago
• yiyixuxu / denoising-diffusion-flax
  Implementing the Denoising Diffusion Probabilistic Model in Flax
  ☆147Updated 2 years ago
• lucidrains / Mega-pytorch
  Implementation of Mega, the Single-head Attention with Multi-headed EMA architecture that currently holds SOTA on Long Range Arena
  ☆204Updated last year
• cloneofsimo / ezmup
  Simple implementation of muP, based on Spectral Condition for Feature Learning. The implementation is SGD only, dont use it for Adam
  ☆73Updated 7 months ago