borisdayma / clip-jaxView external linksLinks
Train vision models using JAX and π€ transformers
β100Dec 14, 2025Updated 2 months ago
Alternatives and similar repositories for clip-jax
Users that are interested in clip-jax are comparing it to the libraries listed below
Sorting:
- β91Sep 19, 2022Updated 3 years ago
- JAX implementation ViT-VQGANβ82Sep 21, 2022Updated 3 years ago
- JAX implementation of VQGANβ91Jul 9, 2022Updated 3 years ago
- Maximal Update Parametrization (ΞΌP) with Flax & Optax.β16Dec 27, 2023Updated 2 years ago
- Implementing the Denoising Diffusion Probabilistic Model in Flaxβ157Nov 1, 2022Updated 3 years ago
- Minimal JAX/Flax port of `lpips` supporting `vgg16`, with pre-trained weights stored in the π€ Hugging Face hub.β17Aug 1, 2022Updated 3 years ago
- HomebrewNLP in JAX flavour for maintable TPU-Trainingβ51Jan 20, 2024Updated 2 years ago
- PIX is an image processing library in JAX, for JAX.β433Mar 6, 2025Updated 11 months ago
- β65Oct 4, 2023Updated 2 years ago
- Implementation of a holodeck, written in Pytorchβ18Nov 1, 2023Updated 2 years ago
- β18Apr 3, 2023Updated 2 years ago
- DataComp: In search of the next generation of multimodal datasetsβ770Apr 28, 2025Updated 9 months ago
- A metrics library for the JAX ecosystemβ40Mar 16, 2023Updated 2 years ago
- Script and models for clustering LAION-400m CLIP embeddings.β26Jan 10, 2022Updated 4 years ago
- Fastai + PyTorch DDP in Jupyter Notebookβ27Sep 29, 2020Updated 5 years ago
- Contrastive Language-Image Pretrainingβ144Sep 6, 2022Updated 3 years ago
- Engineering the state of RNN language models (Mamba, RWKV, etc.)β32May 25, 2024Updated last year
- Using short models to classify long textsβ21Mar 8, 2023Updated 2 years ago
- GATSBI: Generative Adversarial Training for Simulation-Based Inferenceβ19Jul 13, 2023Updated 2 years ago
- This repository shows various ways of deploying a vision model (TensorFlow) from π€ Transformers.β30Aug 22, 2022Updated 3 years ago
- Legible, Scalable, Reproducible Foundation Models with Named Tensors and Jaxβ693Jan 26, 2026Updated 2 weeks ago
- β19Dec 4, 2025Updated 2 months ago
- Easily compute clip embeddings from video framesβ147Oct 31, 2023Updated 2 years ago
- Automatically take good care of your preemptible TPUsβ37May 15, 2023Updated 2 years ago
- Load any clip model with a standardized interfaceβ22Oct 20, 2025Updated 3 months ago
- v objective diffusion inference code for JAX.β215Apr 14, 2022Updated 3 years ago
- β21Jan 5, 2026Updated last month
- Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).β119Jun 5, 2022Updated 3 years ago
- ImageNet-12k subset of ImageNet-21k (fall11)β21Jun 13, 2023Updated 2 years ago
- A minimal Python kernel so you can run Python in your Pythonβ40Apr 17, 2022Updated 3 years ago
- This contains the Flax model of min(DALLΒ·E) and code for converting it to PyTorchβ45Jul 21, 2022Updated 3 years ago
- My attempts at applying Soundstream design on learned tokenization of text and then applying hierarchical attention to text generationβ90Oct 11, 2024Updated last year
- Implementation of the specific Transformer architecture from PaLM - Scaling Language Modeling with Pathways - in Jax (Equinox framework)β190Jun 24, 2022Updated 3 years ago
- β367Apr 12, 2024Updated last year
- CUDA implementation of autoregressive linear attention, with all the latest research findingsβ46May 23, 2023Updated 2 years ago
- JAX implementation of the Llama 2 modelβ216Feb 2, 2024Updated 2 years ago
- A simple, performant and scalable Jax LLM!β2,138Updated this week
- LoRA for arbitrary JAX models and functionsβ145Feb 26, 2024Updated last year
- Pretrained deep learning models for Jax/Flax: StyleGAN2, GPT2, VGG, ResNet, etc.β265Mar 21, 2025Updated 10 months ago