Official JAX implementation of xLSTM including fast and efficient training and inference code. 7B model available at https://huggingface.co/NX-AI/xLSTM-7b.
☆105Apr 8, 2026Updated 5 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.☆94Sep 14, 2026Updated last week
- ☆16Oct 21, 2025Updated 11 months ago
- Official repository of the xLSTM.☆2,204Sep 7, 2026Updated 2 weeks ago
- Dreamer on JAX☆16Jan 19, 2022Updated 4 years ago
- TiRex: Zero-Shot Forecasting Across Long and Short Horizons with Enhanced In-Context Learning☆302Sep 9, 2026Updated last week
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- 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
- Drift-Resilient TabPFN is a method using In-Context Learning via a Prior-Data Fitted Network, to address temporal distribution shifts in …☆30May 17, 2025Updated last year
- Simple repo to finetune an LLM hosted on Hugging Face by creating a LORA☆11Dec 20, 2023Updated 2 years ago
- Implementation of our NeurIPS 2019 paper: Subspace Attack: Exploiting Promising Subspaces for Query-Efficient Black-box Attacks☆10Dec 16, 2019Updated 6 years ago
- MLX implementation of xLSTM model by Beck et al. (2024)☆31Jun 5, 2024Updated 2 years ago
- Explainable Modular Reinforcement Learning (from images)☆16Jan 5, 2026Updated 8 months ago
- xLSTM as Generic Vision Backbone☆490Oct 20, 2025Updated 11 months ago
- Resnets in tensorflow 2.0.☆17Mar 15, 2025Updated last year
- Managed hosting for WordPress and PHP on Cloudways • AdManaged hosting for WordPress, Magento, Laravel, or PHP apps, on multiple cloud providers. Deploy in minutes on Cloudways by DigitalOcean.
- One Initialization to Rule them All: Fine-tuning via Explained Variance Adaptation☆52Oct 20, 2025Updated 11 months ago
- ☆15Jan 16, 2025Updated last year
- Python implementation of fast-oopsi, smc-oopsi, wiener, and discretize with binning☆15May 17, 2022Updated 4 years ago
- the official code of "Diffusion-Based Image-to-Image Translation by Noise Correction via Prompt Interpolation" (ECCV2024)☆13Jan 14, 2025Updated last year
- Code for Krause and Drugowitsch (2022). "A large majority of awake hippocampal sharp-wave ripples feature spatial trajectories with momen…☆16Apr 18, 2023Updated 3 years ago
- ☆36Nov 22, 2024Updated last year
- ☆14Oct 21, 2024Updated last year
- NumPy+Jax with named axes and an uncompromising attitude☆23Mar 4, 2025Updated last year
- ☆10Apr 5, 2024Updated 2 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.
- Convolutive Matrix Factorization in Julia☆10Mar 7, 2021Updated 5 years ago
- Conformal Prediction for Time Series with Modern Hopfield Networks☆85May 28, 2025Updated last year
- HomebrewNLP in JAX flavour for maintable TPU-Training☆50Jan 20, 2024Updated 2 years ago
- Official repository for the MMFM challenge☆26Jun 18, 2024Updated 2 years ago
- Pseudo-Marginal Slice Sampling☆19Jun 8, 2016Updated 10 years ago
- Subset-Norm and Subset-Momentum. This repo is built on top of https://github.com/jiaweizzhao/GaLore.☆19Jul 9, 2025Updated last year
- Machine learning algorithms implements with jax for machine learning in production in large scale dataset.☆23Sep 10, 2026Updated last week
- Official repository for Parallax (Parameterized Local Linear Attention)☆69Jul 30, 2026Updated last month
- Jax SSM Library☆48Nov 24, 2022Updated 3 years ago
- End-to-end encrypted email - Proton Mail • AdSpecial offer: 40% Off Yearly / 80% Off First Month. All Proton services are open source and independently audited for security.
- Official codebase for our paper "Do Language Models Use Their Depth Efficiently?"☆29Jun 25, 2025Updated last year
- ☆19Nov 11, 2025Updated 10 months ago
- Experiments on the impact of depth in transformers and SSMs.☆47Oct 23, 2025Updated 10 months ago
- Programming in Python for Machine Learning - learning Python with an additional focus on Machine Learning (ML)☆63Aug 5, 2021Updated 5 years ago
- Code and data to explore neural scaling laws of xLSTM and Transformer models.☆24Apr 8, 2026Updated 5 months ago
- ☆287Jun 6, 2025Updated last year
- Cross-Country Elimination in JAX☆18Apr 25, 2026Updated 4 months ago