Alternatives and similar repositories for Vision-Transformer-ViT-from-scratch

Users that are interested in Vision-Transformer-ViT-from-scratch are comparing it to the libraries listed below

• tristandeleu / jax-meta-learning
  A collection of meta-learning algorithms in Jax
  ☆23Updated 2 years ago
• KellerJordan / top-sgd
  Optimization algorithm which fits a ResNet to CIFAR-10 5x faster than SGD / Adam (with terrible generalization)
  ☆14Updated last year
• cgarciae / nanoGPT-jax
  The simplest, fastest repository for training/finetuning medium-sized GPTs.
  ☆33Updated last year
• luchris429 / popjaxrl
  Benchmarking RL for POMDPs in Pure JAX [Code for "Structured State Space Models for In-Context Reinforcement Learning" (NeurIPS 2023)]
  ☆103Updated last year
• google-deepmind / PGMax
  Loopy belief propagation for factor graphs on discrete variables in JAX
  ☆152Updated 8 months ago
• fabian-sp / MoMo
  MoMo: Momentum Models for Adaptive Learning Rates
  ☆19Updated last year
• evanatyourservice / psgd_jax
  Implementation of PSGD optimizer in JAX
  ☆33Updated 5 months ago
• phlippe / jax_trainer
  Lightning-like training API for JAX with Flax
  ☆41Updated 6 months ago
• tristandeleu / gfn-maxent-rl
  Comparison between GFlowNets & Maximum Entropy RL
  ☆18Updated last year
• cassidylaidlaw / effective-horizon
  Code and data for the paper "Bridging RL Theory and Practice with the Effective Horizon"
  ☆48Updated 11 months ago
• mohmdelsayed / HesScale
  Scalable Computation of Hessian Diagonals
  ☆13Updated last year
• jax-ml / jax-llm-examples
  ☆126Updated last month
• Hwhitetooth / jax_muzero
  An implementation of MuZero in JAX.
  ☆56Updated 2 years ago
• aidanscannell / iqrl
  iQRL: implicitly Quantized Representations for Sample-efficient Reinforcement Learning
  ☆12Updated 5 months ago
• keraJLi / synthetic-gymnax
  Drop-in environment replacements that make your RL algorithm train faster.
  ☆21Updated last year
• EmptyJackson / groove
  Official implementation of the NeurIPS 2023 paper "Discovering General Reinforcement Learning Algorithms with Adversarial Environment Des…
  ☆28Updated 11 months ago
• josephdviviano / gflownet-tutorials
  Interactive demos describing the main concepts behind GFlowNets
  ☆10Updated last year
• JeremiasKnoblauch / GVIPublic
  Demos for the paper Generalized Variational Inference (Knoblauch, Jewson & Damoulas, 2019)
  ☆20Updated 6 years ago
• DramaCow / jaxued
  ☆81Updated 3 months ago
• lixilinx / psgd_torch
  Pytorch implementation of preconditioned stochastic gradient descent (Kron and affine preconditioner, low-rank approximation precondition…
  ☆177Updated 2 weeks ago
• khdlr / augmax
  Efficiently Composable Data Augmentation on the GPU with Jax
  ☆37Updated last month
• google-research / hyperbo
  Pre-trained Gaussian processes for Bayesian optimization
  ☆92Updated 2 months ago
• bremen79 / parameterfree
  Parameter-Free Optimizers for Pytorch
  ☆130Updated last year
• epignatelli / navix
  Accelerated minigrid environments with JAX
  ☆139Updated 2 weeks ago
• diffprog / code
  Code for the book "The Elements of Differentiable Programming".
  ☆88Updated this week
• probabilisticai / probai-2023
  Materials of the Nordic Probabilistic AI School 2023.
  ☆90Updated last year
• bamos / presentations
  ☆144Updated last month
• google-deepmind / neural_testbed
  ☆192Updated 2 months ago
• y0ast / pytorch-snippets
  Collection of snippets for PyTorch users
  ☆25Updated 3 years ago
• vivekmyers / contrastive_metrics
  Code for the paper "Learning Temporal Distances: Contrastive Successor Features Can Provide a Metric Structure for Decision-Making"
  ☆27Updated 11 months ago