Alternatives and similar repositories for xLSTM-Jax

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

• lucidrains / firefly-torch
  Exploration into the Firefly algorithm in Pytorch
  ☆40Updated 4 months ago
• phlippe / jax_trainer
  Lightning-like training API for JAX with Flax
  ☆41Updated 6 months ago
• AakashKumarNain / mistral_jax
  This is a port of Mistral-7B model in JAX
  ☆32Updated 11 months ago
• yixiaoer / mistral-v0.2-jax
  JAX implementation of the Mistral 7b v0.2 model
  ☆35Updated 11 months ago
• andylolu2 / jax-diffusion
  Implementation of Denoising Diffusion Probabilistic Models (DDPM) in JAX and Flax.
  ☆20Updated last year
• 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 8 months ago
• vvvm23 / mamba-jax
  Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX
  ☆83Updated last year
• radarFudan / mamba-minimal-jax
  ☆31Updated 7 months ago
• lucidrains / frame-averaging-pytorch
  Pytorch implementation of a simple way to enable (Stochastic) Frame Averaging for any network
  ☆50Updated 11 months ago
• crowsonkb / jax-wavelets
  The 2D discrete wavelet transform for JAX
  ☆43Updated 2 years ago
• rdyro / torch2jax
  Wraps PyTorch code in a JIT-compatible way for JAX. Supports automatically defining gradients for reverse-mode AutoDiff.
  ☆54Updated last month
• BirkhoffG / jax-dataloader
  Pytorch-like dataloaders for JAX.
  ☆87Updated last month
• lucidrains / blackbox-gradient-sensing
  Implementation and explorations into Blackbox Gradient Sensing (BGS), an evolutionary strategies approach proposed in a Google Deepmind p…
  ☆16Updated 3 weeks ago
• facebookresearch / adjoint_sampling
  code for "Adjoint Sampling: Highly Scalable Diffusion Samplers via Adjoint Matching"
  ☆99Updated last month
• smonsays / jax-hypernetwork
  A simple hypernetwork implementation in jax using haiku.
  ☆23Updated 2 years ago
• qiauil / torchfsm
  TorchFSM: Fourier Spectral Method with PyTorch
  ☆44Updated 2 weeks ago
• dlwh / jax_sourceror
  Turn jitted jax functions back into python source code
  ☆22Updated 6 months ago
• lucidrains / GAF-microbatch-pytorch
  Implementation of Gradient Agreement Filtering, from Chaubard et al. of Stanford, but for single machine microbatches, in Pytorch
  ☆25Updated 5 months ago
• cgarciae / nnx
  Neural Networks for JAX
  ☆84Updated 9 months ago
• AshishKumar4 / FlaxDiff
  A simple, easy-to-understand library for diffusion models using Flax and Jax. Includes detailed notebooks on DDPM, DDIM, and EDM with sim…
  ☆29Updated last month
• diffprog / code
  Code for the book "The Elements of Differentiable Programming".
  ☆88Updated last week
• dasayan05 / iclr24_blog_code
  Code repo for ICLR 24 BlogPost titled "Building Diffusion Model's theory from ground up"
  ☆20Updated last year
• mfinzi / neural-ivp
  ☆11Updated 2 years ago
• probabilists / azula
  Diffusion models in PyTorch
  ☆104Updated last week
• Artur-Galstyan / statedict2pytree
  ☆43Updated 3 weeks ago
• paganpasta / eqxvision
  A Python package of computer vision models for the Equinox ecosystem.
  ☆107Updated 11 months ago
• lucidrains / holodeck-pytorch
  Implementation of a holodeck, written in Pytorch
  ☆18Updated last year
• google-deepmind / spectral_ssm
  ☆32Updated last year
• joey00072 / microjax
  Jax like function transformation engine but micro, microjax
  ☆32Updated 8 months ago
• itsdaniele / jeometric
  Graph neural networks in JAX.
  ☆67Updated last year