borisdayma / clip-jax
Train vision models using JAX and 🤗 transformers
☆95Updated 3 weeks ago
Related projects ⓘ
Alternatives and complementary repositories for clip-jax
- ☆86Updated 2 years ago
- CLOOB training (JAX) and inference (JAX and PyTorch)☆70Updated 2 years ago
- Automatically take good care of your preemptible TPUs☆32Updated last year
- A JAX implementation of the continuous time formulation of Consistency Models☆83Updated last year
- HomebrewNLP in JAX flavour for maintable TPU-Training☆46Updated 10 months ago
- Latent Diffusion Language Models☆67Updated last year
- ☆18Updated last month
- ☆73Updated 4 months ago
- LoRA for arbitrary JAX models and functions☆132Updated 8 months ago
- Minimal (400 LOC) implementation Maximum (multi-node, FSDP) GPT training☆113Updated 7 months ago
- Utilities for PyTorch distributed☆23Updated last year
- Contrastive Language-Image Pretraining☆143Updated 2 years ago
- JAX implementation ViT-VQGAN☆77Updated 2 years ago
- The 2D discrete wavelet transform for JAX☆38Updated last year
- PyTorch interface for TrueGrad Optimizers☆39Updated last year
- Simple python template☆40Updated 6 months ago
- Implementing the Denoising Diffusion Probabilistic Model in Flax☆142Updated 2 years ago
- Multidimensional indexing for tensors☆113Updated last year
- Serialize JAX, Flax, Haiku, or Objax model params with 🤗`safetensors`☆42Updated 5 months ago
- Fast, Modern, Memory Efficient, and Low Precision PyTorch Optimizers☆58Updated 4 months ago
- Simple implementation of muP, based on Spectral Condition for Feature Learning. The implementation is SGD only, dont use it for Adam☆69Updated 3 months ago
- FID computation in Jax/Flax.☆24Updated 4 months ago
- Mobile Viewer for W&B, built on top of Flutter.☆30Updated 8 months ago
- Efficient optimizers☆79Updated this week
- ☆31Updated 2 months ago
- gpu tester detects broken and slow gpus in a cluster☆67Updated last year
- ☆76Updated 7 months ago
- ☆53Updated 10 months ago