Alternatives and similar repositories for xLSTM-Jax

Users that are interested in xLSTM-Jax are comparing it to the libraries listed below

• vvvm23 / mamba-jax
  Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX
  ☆88Updated last year
• AakashKumarNain / mistral_jax
  This is a port of Mistral-7B model in JAX
  ☆32Updated last year
• lucidrains / frame-averaging-pytorch
  Pytorch implementation of a simple way to enable (Stochastic) Frame Averaging for any network
  ☆51Updated last year
• joey00072 / microjax
  Jax like function transformation engine but micro, microjax
  ☆33Updated 11 months ago
• phlippe / jax_trainer
  Lightning-like training API for JAX with Flax
  ☆43Updated 10 months ago
• google-deepmind / dks
  Multi-framework implementation of Deep Kernel Shaping and Tailored Activation Transformations, which are methods that modify neural netwo…
  ☆74Updated 3 months ago
• matthias-wright / jax-fid
  FID computation in Jax/Flax.
  ☆28Updated last year
• lucidrains / GAF-microbatch-pytorch
  Implementation of Gradient Agreement Filtering, from Chaubard et al. of Stanford, but for single machine microbatches, in Pytorch
  ☆25Updated 8 months ago
• Rose-STL-Lab / AutoSTPP
  Automatic Integration for Neural Spatio-Temporal Point Process models (AI-STPP) is a new paradigm for exact, efficient, non-parametric inf…
  ☆25Updated last year
• lucidrains / kalman-filtering-attention
  Implementation of the Kalman Filtering Attention proposed in "Kalman Filtering Attention for User Behavior Modeling in CTR Prediction"
  ☆59Updated last year
• lucidrains / scaling-vin-pytorch
  Exploration into the Scaling Value Iteration Networks paper, from Schmidhuber's group
  ☆37Updated last year
• ml-gde / jflux
  JAX Implementation of Black Forest Labs' Flux.1 family of models
  ☆39Updated last month
• yixiaoer / mistral-v0.2-jax
  JAX implementation of the Mistral 7b v0.2 model
  ☆35Updated last year
• n2cholas / jax-resnet
  Implementations and checkpoints for ResNet, Wide ResNet, ResNeXt, ResNet-D, and ResNeSt in JAX (Flax).
  ☆114Updated 3 years ago
• teddykoker / learning-to-learn-jax
  JAX implementation of Learning to learn by gradient descent by gradient descent
  ☆27Updated 2 months ago
• google-deepmind / eigengame
  Open source code for EigenGame.
  ☆32Updated 2 years ago
• dylandoblar / noether-networks
  Meta-learning inductive biases in the form of useful conserved quantities.
  ☆37Updated 2 years ago
• google-deepmind / tf2jax
  ☆115Updated last month
• ludwigwinkler / JaxLightning
  Running Jax in PyTorch Lightning
  ☆112Updated 10 months ago
• radarFudan / mamba-minimal-jax
  ☆33Updated 10 months ago
• itsdaniele / jeometric
  Graph neural networks in JAX.
  ☆68Updated last year
• ColinQiyangLi / AdaCat
  AdaCat
  ☆49Updated 3 years ago
• gisilvs / AEF
  ☆33Updated 2 years ago
• attentionneuron / attentionneuron.github.io
  ☆26Updated 3 years ago
• google-deepmind / neural_testbed
  ☆192Updated 3 months ago
• JesseFarebro / flax-mup
  Maximal Update Parametrization (μP) with Flax & Optax.
  ☆16Updated last year
• fattorib / transformer_shmap
  Tensor Parallelism with JAX + Shard Map
  ☆11Updated 2 years ago
• conceptofmind / vit-flax
  Implementation of numerous Vision Transformers in Google's JAX and Flax.
  ☆22Updated 3 years ago
• borisdayma / clip-jax
  Train vision models using JAX and 🤗 transformers
  ☆99Updated last month
• crowsonkb / dice-mc
  DiCE: The Infinitely Differentiable Monte-Carlo Estimator
  ☆31Updated 2 years ago