paper-with-me

홈 › Papers

xLSTM 7B: A Recurrent LLM for Fast and Efficient Inference

2025-03-17 · Maximilian Beck, Korbinian Pöppel, Phillip Lippe, Richard Kurle, Patrick M. Blies, Günter Klambauer, Sebastian Böck, Sepp Hochreiter

Recent breakthroughs in solving reasoning, math and coding problems with Large Language Models (LLMs) have been enabled by investing substantial computation budgets at inference time. Therefore, inference speed is one of the most critical properties of LLM architectures, and there is a growing need for LLMs that are efficient and fast at inference. Recently, LLMs built on the xLSTM architecture have emerged as a powerful alternative to Transformers, offering linear compute scaling with sequence length and constant memory usage, both highly desirable properties for efficient inference. However, such xLSTM-based LLMs have yet to be scaled to larger models and assessed and compared with respect to inference speed and efficiency. In this work, we introduce xLSTM 7B, a 7-billion-parameter LLM that combines xLSTM's architectural benefits with targeted optimizations for fast and efficient inference. Our experiments demonstrate that xLSTM 7B achieves performance on downstream tasks comparable to other similar-sized LLMs, while providing significantly faster inference speeds and greater efficiency compared to Llama- and Mamba-based LLMs. These results establish xLSTM 7B as the fastest and most efficient 7B LLM, offering a solution for tasks that require large amounts of test-time computation. Our work highlights xLSTM's potential as a foundational architecture for methods building on heavy use of LLM inference. Our model weights, model code and training code are open-source.

📄 PDF Abstract BibTeX arXiv:2503.13427

Code (3)

nx-ai/mlstm_kernels 공식 구현 jax
nx-ai/xlstm 공식 구현 pytorch
nx-ai/xlstm-jax 공식 구현 jax

Tasks

MambaMath

Methods 이 논문이 사용한 방법론

SPEED The monocular depth estimation (MDE) is the task of estimating depth from a single frame. This information is an essential knowledge in many computer vision tasks such as scene…

Similar Papers 제목 키워드 기반

A Large Recurrent Action Model: xLSTM enables Fast Inference for Robotics Tasks

2024-10-29 · Thomas Schmied, Thomas Adler, Vihang Patil, Maximilian Beck 외

In recent years, there has been a trend in the field of Reinforcement Learning (RL) towards large action models trained offline on large-scale datasets via sequence modeling. Existing models are primarily based on the Tr…

MambaReinforcement Learning (RL)

Distil-xLSTM: Learning Attention Mechanisms through Recurrent Structures

2025-03-24 · Abdoul Majid O. Thiombiano, Brahim Hnich, Ali Ben Mrad, Mohamed Wiem Mkaouer

The current era of Natural Language Processing (NLP) is dominated by Transformer models. However, novel architectures relying on recurrent mechanisms, such as xLSTM and Mamba, have been proposed as alternatives to attent…

Language ModelingLanguage ModellingLarge Language ModelMamba+1

Linear-Time and Constant-Memory Text Embeddings Based on Recurrent Language Models

2026-04-20 · Tobias Grantner, Emanuel Sallinger, Martin Flechl arxiv

Transformer-based embedding models suffer from quadratic computational and linear memory complexity, limiting their utility for long sequences. We propose recurrent architectures as an efficient alternative, introducing …

Bio-xLSTM: Generative modeling, representation and in-context learning of biological and chemical sequences

2024-11-06 · Niklas Schmidinger, Lisa Schneckenreiter, Philipp Seidl, Johannes Schimunek 외

Language models for biological and chemical sequences enable crucial applications such as drug discovery, protein engineering, and precision medicine. Currently, these language models are predominantly based on Transform…

Drug DiscoveryIn-Context Learning

StoxLSTM: A Stochastic Extended Long Short-Term Memory Network for Time Series Forecasting

2025-09-01 · Zihao Wang, Yunjie Li, Lingmin Zan, Zheng Gong 외 arxiv

The Extended Long Short-Term Memory (xLSTM) network has demonstrated strong capability in modeling complex long-term dependencies in time series data. Despite its success, the deterministic architecture of xLSTM limits i…

Time Series Forecasting