Alternatives and similar repositories for jax-vqvae-gpt:

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

• lucidrains / token-shift-gpt
  Implementation of Token Shift GPT - An autoregressive model that solely relies on shifting the sequence space for mixing
  ☆48Updated 3 years ago
• srush / mamba-scans
  Blog post
  ☆17Updated last year
• ClashLuke / tpucare
  Automatically take good care of your preemptible TPUs
  ☆36Updated last year
• EleutherAI / rnngineering
  Engineering the state of RNN language models (Mamba, RWKV, etc.)
  ☆32Updated 10 months ago
• conceptofmind / vit-flax
  Implementation of numerous Vision Transformers in Google's JAX and Flax.
  ☆22Updated 2 years ago
• radarFudan / mamba-minimal-jax
  ☆30Updated 4 months ago
• toshas / torch-discounted-cumsum
  Fast Discounted Cumulative Sums in PyTorch
  ☆95Updated 3 years ago
• lucaslingle / mu_transformer
  Transformer with Mu-Parameterization, implemented in Jax/Flax. Supports FSDP on TPU pods.
  ☆30Updated 3 months ago
• ahennequ / pytorch-custom-mma
  ☆29Updated 2 years ago
• vvvm23 / mamba-jax
  Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX
  ☆83Updated last year
• peerdavid / layerwise-batch-entropy
  Layerwise Batch Entropy Regularization
  ☆22Updated 2 years ago
• Z-T-WANG / LaProp-Optimizer
  Codes accompanying the paper "LaProp: a Better Way to Combine Momentum with Adaptive Gradient"
  ☆28Updated 4 years ago
• RobertCsordas / moe
  Official repository for the paper "Approximating Two-Layer Feedforward Networks for Efficient Transformers"
  ☆36Updated last year
• ColinQiyangLi / AdaCat
  AdaCat
  ☆49Updated 2 years ago
• matthias-wright / jax-fid
  FID computation in Jax/Flax.
  ☆27Updated 8 months ago
• sholtodouglas / multihost_dataloading
  Experimenting with how best to do multi-host dataloading
  ☆10Updated 2 years ago
• young-geng / mlxu
  Machine Learning eXperiment Utilities
  ☆46Updated 9 months ago
• google-deepmind / spectral_ssm
  ☆31Updated 11 months ago
• edwardmilsom / function-space-learning-rates-paper
  Code for the paper "Function-Space Learning Rates"
  ☆17Updated last month
• HomebrewML / Olmax
  HomebrewNLP in JAX flavour for maintable TPU-Training
  ☆49Updated last year
• google-research / precondition
  ☆31Updated last week
• dylandoblar / noether-networks
  Meta-learning inductive biases in the form of useful conserved quantities.
  ☆37Updated 2 years ago
• amirzandieh / HyperAttention
  Triton Implementation of HyperAttention Algorithm
  ☆47Updated last year
• ztjhz / t5-jax
  JAX implementation of the T5 model: Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformer
  ☆22Updated last year
• teddykoker / tinyloader
  ☆63Updated last week
• RobertCsordas / moe_layer
  sigma-MoE layer
  ☆18Updated last year
• hr0nix / dejax
  Accelerated replay buffers in JAX
  ☆41Updated 2 years ago
• srush / drop7
  ☆18Updated 11 months ago
• crowsonkb / dice-mc
  DiCE: The Infinitely Differentiable Monte-Carlo Estimator
  ☆31Updated last year
• teddykoker / learning-to-learn-jax
  JAX implementation of Learning to learn by gradient descent by gradient descent
  ☆27Updated 5 months ago