Official JAX implementation of xLSTM including fast and efficient training and inference code. 7B model available at https://huggingface.co/NX-AI/xLSTM-7b.
☆105Jan 8, 2025Updated last year
Alternatives and similar repositories for xlstm-jax
Users that are interested in xlstm-jax are comparing it to the libraries listed below
Sorting:
- Official repository of the xLSTM.☆2,125Nov 4, 2025Updated 4 months ago
- a minimal website to get the diff of llm rewrites☆11Dec 11, 2024Updated last year
- Drift-Resilient TabPFN is a method using In-Context Learning via a Prior-Data Fitted Network, to address temporal distribution shifts in …☆28May 17, 2025Updated 10 months ago
- TiRex: Zero-Shot Forecasting Across Long and Short Horizons with Enhanced In-Context Learning☆257Mar 4, 2026Updated 2 weeks ago
- One Initialization to Rule them All: Fine-tuning via Explained Variance Adaptation☆49Oct 20, 2025Updated 5 months ago
- Jax/Flax implementation of DeiT and DeiT-III (ViT)☆19Dec 21, 2024Updated last year
- Plotly in FastHTML☆26Aug 10, 2025Updated 7 months ago
- ☆18Nov 11, 2025Updated 4 months ago
- Optimised Extended LSTM for time-series forecasting☆40Nov 16, 2024Updated last year
- MLX implementation of xLSTM model by Beck et al. (2024)☆32Jun 5, 2024Updated last year
- xLSTM as Generic Vision Backbone☆491Oct 20, 2025Updated 5 months ago
- Functional matrix factorization via Bayesian tensor filtering☆13Oct 1, 2025Updated 5 months ago
- ☆30Feb 27, 2024Updated 2 years ago
- Resnets in tensorflow 2.0.☆17Mar 15, 2025Updated last year
- Interactive timeline visualization for Org-roam.☆36Jan 12, 2026Updated 2 months ago
- ☆10Jan 12, 2024Updated 2 years ago
- Adapter-X: A Novel General Parameter-Efficient Fine-Tuning Framework for Vision☆11Jul 22, 2024Updated last year
- Reasoning Activation in LLMs via Small Model Transfer (NeurIPS 2025)☆22Oct 16, 2025Updated 5 months ago
- Code for Research Project TLDR☆25Jul 28, 2025Updated 7 months ago
- ☆259Jun 6, 2025Updated 9 months ago
- Control inverted pendulum by LQR in OpenAI Gym☆12Oct 2, 2024Updated last year
- Official repository for Activation-Informed Merging (AIM) of Large Language Models☆21Feb 10, 2025Updated last year
- Implementation of Rational Kolmogorov-Arnold Network (rKAN)☆20Jun 28, 2024Updated last year
- ☆35Jul 16, 2025Updated 8 months ago
- This is the open-source code for TokenCarve.☆24Jan 23, 2026Updated last month
- Efficient encoder-decoder architecture for small language models (≤1B parameters) with cross-architecture knowledge distillation and visi…☆32Feb 7, 2025Updated last year
- ☆13Jan 16, 2025Updated last year
- ☆14Oct 21, 2024Updated last year
- Julia implementation of flash-attention operation for neural networks.☆11May 31, 2023Updated 2 years ago
- Conformal Prediction for Time Series with Modern Hopfield Networks☆87May 28, 2025Updated 9 months ago
- ☆35Nov 22, 2024Updated last year
- NumPy+Jax with named axes and an uncompromising attitude☆23Mar 4, 2025Updated last year
- Flow-matching algorithms in JAX☆116Aug 12, 2024Updated last year
- Einsum-like high-level array sharding API for JAX☆34Jul 16, 2024Updated last year
- Efficient and Effective Weight-Ensembling Mixture of Experts for Multi-Task Model Merging. Arxiv, 2024.☆16Oct 28, 2024Updated last year
- Pseudo-Marginal Slice Sampling☆19Jun 8, 2016Updated 9 years ago
- Code for Krause and Drugowitsch (2022). "A large majority of awake hippocampal sharp-wave ripples feature spatial trajectories with momen…☆15Apr 18, 2023Updated 2 years ago
- Jax SSM Library☆48Nov 24, 2022Updated 3 years ago
- Numerical Optimisation Library☆17Jul 9, 2023Updated 2 years ago