lucidrains / DALLE2-jax

Related projects: ⓘ

• borisdayma / clip-jax
  Train vision models using JAX and 🤗 transformers
  ☆91Updated last week
• kingoflolz / CLIP_JAX
  Contrastive Language-Image Pretraining
  ☆140Updated 2 years ago
• patil-suraj / stable-diffusion-jax
  ☆85Updated 2 years ago
• crowsonkb / jax-wavelets
  The 2D discrete wavelet transform for JAX
  ☆36Updated last year
• crowsonkb / esgd
  ESGD-M is a stochastic non-convex second order optimizer, suitable for training deep learning models, for PyTorch.
  ☆56Updated 2 years ago
• HomebrewNLP / Olmax
  HomebrewNLP in JAX flavour for maintable TPU-Training
  ☆46Updated 7 months ago
• TomFrederik / grokking
  Re-implementation of 'Grokking: Generalization beyond overfitting on small algorithmic datasets'
  ☆38Updated 2 years ago
• unixpickle / vae-textures
  Texture mapping with variational auto-encoders
  ☆40Updated 2 years ago
• crowsonkb / cloob-training
  CLOOB training (JAX) and inference (JAX and PyTorch)
  ☆70Updated 2 years ago
• matthias-wright / jax-fid
  FID computation in Jax/Flax.
  ☆23Updated 2 months ago
• 4rtemi5 / imax
  Image augmentation library for Jax
  ☆36Updated 5 months ago
• n2cholas / jax-resnet
  Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).
  ☆103Updated 2 years ago
• ClashLuke / TrueGrad
  PyTorch interface for TrueGrad Optimizers
  ☆39Updated last year
• gisilvs / AEF
  ☆33Updated last year
• lucidrains / mlp-gpt-jax
  A GPT, made only of MLPs, in Jax
  ☆55Updated 3 years ago
• rolandgvc / flaxvision
  A selection of neural network models ported from torchvision for JAX & Flax.
  ☆44Updated 3 years ago
• lucidrains / poolformer
  ☆40Updated this week
• patil-suraj / vit-vqgan
  JAX implementation ViT-VQGAN
  ☆77Updated last year
• lucidrains / hourglass-transformer-pytorch
  Implementation of Hourglass Transformer, in Pytorch, from Google and OpenAI
  ☆74Updated 2 years ago
• Jack000 / DALLE-pytorch
  Implementation / replication of DALL-E, OpenAI's Text to Image Transformer, in Pytorch
  ☆90Updated 2 years ago
• ColinQiyangLi / AdaCat
  AdaCat
  ☆49Updated 2 years ago
• alvarobartt / safejax
  Serialize JAX, Flax, Haiku, or Objax model params with 🤗`safetensors`
  ☆41Updated 3 months ago
• crowsonkb / consistency-models
  A JAX implementation of the continuous time formulation of Consistency Models
  ☆83Updated last year
• ClashLuke / tpucare
  Automatically take good care of your preemptible TPUs
  ☆28Updated last year
• arogozhnikov / eindex
  Multidimensional indexing for tensors
  ☆108Updated last year
• shawwn / ml-notes
  ☆38Updated last year
• j-towns / vdvae-jax
  Very deep VAEs in JAX/Flax
  ☆45Updated 3 years ago
• n2cholas / progan-flax
  Flax (JAX) implementation of Progressive Growing of GANs for Improved Quality, Stability, and Variation
  ☆12Updated 3 years ago
• NTT123 / pax
  A stateful pytree library for training neural networks.
  ☆21Updated 2 years ago
• lucidrains / cogvideo-pytorch
  ☆38Updated this week