Alternatives and similar repositories for xLSTM-Jax:

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

• mfinzi / neural-ivp
  ☆11Updated last year
• lucidrains / firefly-torch
  Exploration into the Firefly algorithm in Pytorch
  ☆38Updated 2 months ago
• AakashKumarNain / mistral_jax
  This is a port of Mistral-7B model in JAX
  ☆32Updated 10 months ago
• vvvm23 / mamba-jax
  Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX
  ☆83Updated last year
• crowsonkb / jax-wavelets
  The 2D discrete wavelet transform for JAX
  ☆43Updated 2 years ago
• phlippe / jax_trainer
  Lightning-like training API for JAX with Flax
  ☆38Updated 5 months ago
• dylandoblar / noether-networks
  Meta-learning inductive biases in the form of useful conserved quantities.
  ☆37Updated 2 years ago
• rdyro / torch2jax
  Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.
  ☆52Updated last week
• lucidrains / GAF-microbatch-pytorch
  Implementation of Gradient Agreement Filtering, from Chaubard et al. of Stanford, but for single machine microbatches, in Pytorch
  ☆24Updated 3 months ago
• NTT123 / pax
  A stateful pytree library for training neural networks.
  ☆22Updated 2 years ago
• dasayan05 / iclr24_blog_code
  Code repo for ICLR 24 BlogPost titled "Building Diffusion Model's theory from ground up"
  ☆18Updated last year
• layer6ai-labs / calo-forest
  A scalable implementation of diffusion and flow-matching with XGBoost models, applied to calorimeter data.
  ☆18Updated 6 months ago
• itsdaniele / jeometric
  Graph neural networks in JAX.
  ☆67Updated 10 months ago
• joey00072 / microjax
  Jax like function transformation engine but micro, microjax
  ☆31Updated 6 months ago
• google-deepmind / tf2jax
  ☆114Updated last week
• BirkhoffG / jax-dataloader
  Pytorch-like dataloaders for JAX.
  ☆81Updated last week
• 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…
  ☆24Updated 6 months ago
• sayakpaul / BiT-jax2tf
  This repository hosts the code to port NumPy model weights of BiT-ResNets to TensorFlow SavedModel format.
  ☆14Updated 3 years ago
• matthias-wright / jax-fid
  FID computation in Jax/Flax.
  ☆27Updated 9 months ago
• smonsays / metax
  flexible meta-learning in jax
  ☆13Updated last year
• cgarciae / nnx
  Neural Networks for JAX
  ☆84Updated 7 months ago
• srush / anynp
  Proof-of-concept of global switching between numpy/jax/pytorch in a library.
  ☆18Updated 10 months ago
• fattorib / transformer_shmap
  Tensor Parallelism with JAX + Shard Map
  ☆11Updated last year
• srush / Tensor-Puzzles-Penzai
  ☆20Updated last year
• alvarobartt / safejax
  Serialize JAX, Flax, Haiku, or Objax model params with 🤗`safetensors`
  ☆44Updated 11 months ago
• evanatyourservice / psgd_jax
  Implementation of PSGD optimizer in JAX
  ☆33Updated 4 months ago
• andylolu2 / jax-diffusion
  Implementation of Denoising Diffusion Probabilistic Models (DDPM) in JAX and Flax.
  ☆20Updated last year
• google-deepmind / spectral_ssm
  ☆31Updated last year
• ludwigwinkler / JaxLightning
  Running Jax in PyTorch Lightning
  ☆99Updated 4 months ago
• 4rtemi5 / imax
  Image augmentation library for Jax
  ☆39Updated last year