Official JAX implementation of xLSTM including fast and efficient training and inference code. 7B model available at https://huggingface.co/NX-AI/xLSTM-7b.
☆106Apr 8, 2026Updated 2 months ago
Alternatives and similar repositories for xlstm-jax
Users that are interested in xlstm-jax are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Tiled Flash Linear Attention library for fast and efficient mLSTM Kernels.☆89Mar 27, 2026Updated 2 months ago
- FlashRNN - Fast RNN Kernels with I/O Awareness☆182Oct 20, 2025Updated 7 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 …☆29May 17, 2025Updated last year
- Dreamer on JAX☆16Jan 19, 2022Updated 4 years ago
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click. Zero configuration with optimized deployments.
- Partially Observable Multi-Agent RL with Transformers☆17Apr 27, 2026Updated last month
- One Initialization to Rule them All: Fine-tuning via Explained Variance Adaptation☆53Oct 20, 2025Updated 7 months ago
- ☆12Dec 4, 2024Updated last year
- Jax/Flax implementation of DeiT and DeiT-III (ViT)☆19Dec 21, 2024Updated last year
- MetaArcade is a configurable environment suite for meta-learning☆16Oct 19, 2022Updated 3 years ago
- Mamba4Cast, a zero-shot time series forecasting model, achieves competitive performance and faster inference than transformer-based model…☆54Oct 11, 2024Updated last year
- Simple repo to finetune an LLM hosted on Hugging Face by creating a LORA☆11Dec 20, 2023Updated 2 years ago
- Optimised Extended LSTM for time-series forecasting☆44May 2, 2026Updated last month
- ☆10Nov 13, 2024Updated last year
- GPUs on demand by Runpod - Special Offer Available • AdRun AI, ML, and HPC workloads on powerful cloud GPUs—without limits or wasted spend. Deploy GPUs in under a minute and pay by the second.
- [TMLR 2024] Official PyTorch Implementation of Deep Dynamic Latent Particles☆18Mar 25, 2026Updated 2 months ago
- Resnets in tensorflow 2.0.☆17Mar 15, 2025Updated last year
- ☆30Feb 27, 2024Updated 2 years ago
- Meta-Prompting for Automating Zero-shot Visual Recognition with LLMs (ECCV 2024)☆19Jul 15, 2024Updated last year
- Vision-Language Guidance for LiDAR-based Unsupervised 3D Object Detection☆27Nov 21, 2024Updated last year
- Code and data for paper "(How) do Language Models Track State?"☆24Mar 31, 2025Updated last year
- The implementation for FREE-Merging: Fourier Transform for Model Merging with Lightweight Experts (ICCV25)☆15Jun 26, 2025Updated 11 months ago
- Adapter-X: A Novel General Parameter-Efficient Fine-Tuning Framework for Vision☆11Jul 22, 2024Updated last year
- ☆260Jun 6, 2025Updated last year
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click. Zero configuration with optimized deployments.
- Reasoning Activation in LLMs via Small Model Transfer (NeurIPS 2025)☆22Oct 16, 2025Updated 7 months ago
- Implementation of Rational Kolmogorov-Arnold Network (rKAN)☆21Jun 28, 2024Updated last year
- The official Languini Kitchen repository☆14May 6, 2024Updated 2 years ago
- Efficient encoder-decoder architecture for small language models (≤1B parameters) with cross-architecture knowledge distillation and visi…☆32Feb 7, 2025Updated last year
- ☆29Jan 8, 2026Updated 5 months ago
- ☆13Jan 16, 2025Updated last year
- ☆36Nov 22, 2024Updated last year
- Conformal Prediction for Time Series with Modern Hopfield Networks☆86May 28, 2025Updated last year
- Julia implementation of flash-attention operation for neural networks.☆11May 31, 2023Updated 3 years ago
- GPU virtual machines on DigitalOcean Gradient AI • AdGet to production fast with high-performance AMD and NVIDIA GPUs you can spin up in seconds. The definition of operational simplicity.
- Heatmap text in Julia.☆11Jul 3, 2025Updated 11 months ago
- HomebrewNLP in JAX flavour for maintable TPU-Training☆51Jan 20, 2024Updated 2 years ago
- NumPy+Jax with named axes and an uncompromising attitude☆23Mar 4, 2025Updated last year
- Einsum-like high-level array sharding API for JAX☆34Jul 16, 2024Updated last year
- Flow-matching algorithms in JAX☆117Aug 12, 2024Updated last year
- Efficient and Effective Weight-Ensembling Mixture of Experts for Multi-Task Model Merging. Arxiv, 2024.☆16Oct 28, 2024Updated last year
- Code for Krause and Drugowitsch (2022). "A large majority of awake hippocampal sharp-wave ripples feature spatial trajectories with momen…☆15Apr 18, 2023Updated 3 years ago