Alternatives and similar repositories for clip-jax

Users that are interested in clip-jax are comparing it to the libraries listed below

• patil-suraj / stable-diffusion-jax

  View on GitHub
  ☆91Sep 19, 2022Updated 3 years ago
• patil-suraj / vit-vqgan

  View on GitHub
  JAX implementation ViT-VQGAN
  ☆82Sep 21, 2022Updated 3 years ago
• patil-suraj / vqgan-jax

  View on GitHub
  JAX implementation of VQGAN
  ☆91Jul 9, 2022Updated 3 years ago
• JesseFarebro / flax-mup

  View on GitHub
  Maximal Update Parametrization (μP) with Flax & Optax.
  ☆16Dec 27, 2023Updated 2 years ago
• yiyixuxu / denoising-diffusion-flax

  View on GitHub
  Implementing the Denoising Diffusion Probabilistic Model in Flax
  ☆157Nov 1, 2022Updated 3 years ago
• pcuenca / lpips-j

  View on GitHub
  Minimal JAX/Flax port of `lpips` supporting `vgg16`, with pre-trained weights stored in the 🤗 Hugging Face hub.
  ☆17Aug 1, 2022Updated 3 years ago
• HomebrewML / Olmax

  View on GitHub
  HomebrewNLP in JAX flavour for maintable TPU-Training
  ☆51Jan 20, 2024Updated 2 years ago
• google-deepmind / dm_pix

  View on GitHub
  PIX is an image processing library in JAX, for JAX.
  ☆433Mar 6, 2025Updated 11 months ago
• LAION-AI / General-GPT

  View on GitHub
  ☆65Oct 4, 2023Updated 2 years ago
• lucidrains / holodeck-pytorch

  View on GitHub
  Implementation of a holodeck, written in Pytorch
  ☆18Nov 1, 2023Updated 2 years ago
• johnrobinsn / alpaca_lora_30b_4bit

  View on GitHub
  ☆18Apr 3, 2023Updated 2 years ago
• mlfoundations / datacomp

  View on GitHub
  DataComp: In search of the next generation of multimodal datasets
  ☆770Apr 28, 2025Updated 9 months ago
• cgarciae / jax_metrics

  View on GitHub
  A metrics library for the JAX ecosystem
  ☆40Mar 16, 2023Updated 2 years ago
• pbaylies / clustering-laion400m

  View on GitHub
  Script and models for clustering LAION-400m CLIP embeddings.
  ☆26Jan 10, 2022Updated 4 years ago
• philtrade / Ddip

  View on GitHub
  Fastai + PyTorch DDP in Jupyter Notebook
  ☆27Sep 29, 2020Updated 5 years ago
• kingoflolz / CLIP_JAX

  View on GitHub
  Contrastive Language-Image Pretraining
  ☆144Sep 6, 2022Updated 3 years ago
• EleutherAI / rnngineering

  View on GitHub
  Engineering the state of RNN language models (Mamba, RWKV, etc.)
  ☆32May 25, 2024Updated last year
• nbroad1881 / strideformer

  View on GitHub
  Using short models to classify long texts
  ☆21Mar 8, 2023Updated 2 years ago
• mackelab / gatsbi

  View on GitHub
  GATSBI: Generative Adversarial Training for Simulation-Based Inference
  ☆19Jul 13, 2023Updated 2 years ago
• sayakpaul / deploy-hf-tf-vision-models

  View on GitHub
  This repository shows various ways of deploying a vision model (TensorFlow) from 🤗 Transformers.
  ☆30Aug 22, 2022Updated 3 years ago
• marin-community / levanter

  View on GitHub
  Legible, Scalable, Reproducible Foundation Models with Named Tensors and Jax
  ☆693Jan 26, 2026Updated 2 weeks ago
• ethansmith2000 / TransformerExperiments

  View on GitHub
  ☆19Dec 4, 2025Updated 2 months ago
• iejMac / clip-video-encode

  View on GitHub
  Easily compute clip embeddings from video frames
  ☆147Oct 31, 2023Updated 2 years ago
• ClashLuke / tpucare

  View on GitHub
  Automatically take good care of your preemptible TPUs
  ☆37May 15, 2023Updated 2 years ago
• data2ml / all-clip

  View on GitHub
  Load any clip model with a standardized interface
  ☆22Oct 20, 2025Updated 3 months ago
• crowsonkb / v-diffusion-jax

  View on GitHub
  v objective diffusion inference code for JAX.
  ☆215Apr 14, 2022Updated 3 years ago
• huggingface / hf-endpoints-documentation

  View on GitHub
  ☆21Jan 5, 2026Updated last month
• n2cholas / jax-resnet

  View on GitHub
  Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).
  ☆119Jun 5, 2022Updated 3 years ago
• rwightman / imagenet-12k

  View on GitHub
  ImageNet-12k subset of ImageNet-21k (fall11)
  ☆21Jun 13, 2023Updated 2 years ago
• fastai / tinykernel

  View on GitHub
  A minimal Python kernel so you can run Python in your Python
  ☆40Apr 17, 2022Updated 3 years ago
• kuprel / min-dalle-flax

  View on GitHub
  This contains the Flax model of min(DALL·E) and code for converting it to PyTorch
  ☆45Jul 21, 2022Updated 3 years ago
• lucidrains / rvq-vae-gpt

  View on GitHub
  My attempts at applying Soundstream design on learned tokenization of text and then applying hierarchical attention to text generation
  ☆90Oct 11, 2024Updated last year
• lucidrains / PaLM-jax

  View on GitHub
  Implementation of the specific Transformer architecture from PaLM - Scaling Language Modeling with Pathways - in Jax (Equinox framework)
  ☆190Jun 24, 2022Updated 3 years ago
• google / flaxformer

  View on GitHub
  ☆367Apr 12, 2024Updated last year
• lucidrains / autoregressive-linear-attention-cuda

  View on GitHub
  CUDA implementation of autoregressive linear attention, with all the latest research findings
  ☆46May 23, 2023Updated 2 years ago
• ayaka14732 / llama-2-jax

  View on GitHub
  JAX implementation of the Llama 2 model
  ☆216Feb 2, 2024Updated 2 years ago
• AI-Hypercomputer / maxtext

  View on GitHub
  A simple, performant and scalable Jax LLM!
  ☆2,138Updated this week
• davisyoshida / lorax

  View on GitHub
  LoRA for arbitrary JAX models and functions
  ☆145Feb 26, 2024Updated last year
• matthias-wright / flaxmodels

  View on GitHub
  Pretrained deep learning models for Jax/Flax: StyleGAN2, GPT2, VGG, ResNet, etc.
  ☆265Mar 21, 2025Updated 10 months ago