conceptofmind / vit-flax

Related projects ⓘ

Alternatives and complementary repositories for vit-flax

• matthias-wright / jax-fid
  FID computation in Jax/Flax.
  ☆24Updated 4 months ago
• n2cholas / progan-flax
  Flax (JAX) implementation of Progressive Growing of GANs for Improved Quality, Stability, and Variation
  ☆12Updated 3 years ago
• sayakpaul / big_vision_experiments
  Contains my experiments with the `big_vision` repo to train ViTs on ImageNet-1k.
  ☆22Updated last year
• FrancescoSaverioZuppichini / Loading-huge-PyTorch-models-with-linear-memory-consumption
  Little article showing how to load pytorch's models with linear memory consumption
  ☆34Updated 2 years ago
• borisdayma / clip-jax
  Train vision models using JAX and 🤗 transformers
  ☆95Updated last month
• lucidrains / frame-averaging-pytorch
  Pytorch implementation of a simple way to enable (Stochastic) Frame Averaging for any network
  ☆48Updated 3 months ago
• lucidrains / pytorch-custom-utils
  Just some miscellaneous utility functions / decorators / modules related to Pytorch and Accelerate to help speed up implementation of new…
  ☆119Updated 3 months ago
• lucidrains / gateloop-transformer
  Implementation of GateLoop Transformer in Pytorch and Jax
  ☆86Updated 5 months ago
• radarFudan / mamba-minimal-jax
  ☆29Updated 2 months ago
• vballoli / vit-flax
  Implementation of Vision Transformers in Flax
  ☆17Updated 4 years ago
• lucidrains / scaling-vin-pytorch
  Exploration into the Scaling Value Iteration Networks paper, from Schmidhuber's group
  ☆36Updated last month
• lucidrains / einops-exts
  Implementation of some personal helper functions for Einops, my most favorite tensor manipulation library ❤️
  ☆52Updated last year
• lucidrains / token-shift-gpt
  Implementation of Token Shift GPT - An autoregressive model that solely relies on shifting the sequence space for mixing
  ☆47Updated 2 years ago
• AakashKumarNain / mistral_jax
  This is a port of Mistral-7B model in JAX
  ☆30Updated 4 months ago
• enajx / HyperNCA
  ☆36Updated 2 years ago
• lucidrains / perceiver-ar-pytorch
  Implementation of Perceiver AR, Deepmind's new long-context attention network based on Perceiver architecture, in Pytorch
  ☆86Updated last year
• lucidrains / kalman-filtering-attention
  Implementation of the Kalman Filtering Attention proposed in "Kalman Filtering Attention for User Behavior Modeling in CTR Prediction"
  ☆57Updated last year
• Zasder3 / open_clip_juwels
  An open source implementation of CLIP.
  ☆32Updated 2 years ago
• crowsonkb / torch-dist-utils
  Utilities for PyTorch distributed
  ☆23Updated last year
• lucidrains / spline-based-transformer
  Implementation of the proposed Spline-Based Transformer from Disney Research
  ☆76Updated 2 weeks ago
• sayakpaul / BiT-jax2tf
  This repository hosts the code to port NumPy model weights of BiT-ResNets to TensorFlow SavedModel format.
  ☆14Updated 2 years ago
• jiaweizzhao / ZerO-initialization
  ☆73Updated 2 years ago
• nikhilvyas / SOAP
  ☆129Updated last week
• lucidrains / autoregressive-linear-attention-cuda
  CUDA implementation of autoregressive linear attention, with all the latest research findings
  ☆43Updated last year
• patil-suraj / vit-vqgan
  JAX implementation ViT-VQGAN
  ☆77Updated 2 years ago
• tanelp / confviews
  Generate bird's-eye views of conference proceedings.
  ☆22Updated 4 months ago
• kingoflolz / CLIP_JAX
  Contrastive Language-Image Pretraining
  ☆143Updated 2 years ago
• phlippe / jax_trainer
  Lightning-like training API for JAX with Flax
  ☆34Updated 6 months ago