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 6 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.☆96Sep 14, 2026Updated 3 weeks ago
- FlashRNN - Fast RNN Kernels with I/O Awareness☆190Sep 14, 2026Updated 3 weeks ago
- ☆23Aug 19, 2026Updated last month
- a minimal website to get the diff of llm rewrites☆11Dec 11, 2024Updated last year
- Dreamer on JAX☆16Jan 19, 2022Updated 4 years ago
- 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.
- Jax/Flax implementation of DeiT and DeiT-III (ViT)☆19Dec 21, 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 …☆31May 17, 2025Updated last year
- ☆12Dec 4, 2024Updated last year
- Simple repo to finetune an LLM hosted on Hugging Face by creating a LORA☆11Dec 20, 2023Updated 2 years ago
- [ICCV 2025] Code for "Flow Stochastic Segmentation Networks"☆19Jun 9, 2026Updated 4 months ago
- ☆10Nov 13, 2024Updated last year
- MLX implementation of xLSTM model by Beck et al. (2024)☆31Jun 5, 2024Updated 2 years ago
- Launching and monitoring Slurm experiments in Python☆35Jun 29, 2026Updated 3 months ago
- Code of the paper "Interactive Learning of Temporal Feature for Control", published in the IEEE Robotics & Automation Magazine.☆12Dec 27, 2022Updated 3 years ago
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- VisualOverload (CVPR 2026) is a VQA benchmark for image understanding in dense, high-resolution scenes.☆18May 31, 2026Updated 4 months ago
- ☆15Updated this week
- Benchmarks of all public available SNARK/STARK keccak circuits☆13Oct 1, 2023Updated 3 years ago
- ☆260Jun 6, 2025Updated last year
- The official Languini Kitchen repository☆14May 6, 2024Updated 2 years ago
- ☆29Jan 8, 2026Updated 9 months ago
- ☆36Nov 22, 2024Updated last year
- Adapter-X: A Novel General Parameter-Efficient Fine-Tuning Framework for Vision☆11Jul 22, 2024Updated 2 years ago
- ☆14Oct 21, 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.
- NumPy+Jax with named axes and an uncompromising attitude☆23Mar 4, 2025Updated last year
- Official implementation of the ICML 2025 paper "SOLD: Slot Object-Centric Latent Dynamics Models for Relational Manipulation Learning fro…☆23Sep 30, 2025Updated last year
- Julia implementation of flash-attention operation for neural networks.☆11May 31, 2023Updated 3 years ago
- HomebrewNLP in JAX flavour for maintable TPU-Training☆50Jan 20, 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.☆25Oct 1, 2026Updated last week
- Machine Learning Methods for Neural Data Analysis☆22May 16, 2025Updated last year
- Official codebase for our paper "Do Language Models Use Their Depth Efficiently?"☆30Jun 25, 2025Updated last year
- Wordpress hosting with auto-scaling - Free Trial Offer • AdFully Managed hosting for WordPress and WooCommerce businesses that need reliable, auto-scalable performance. Cloudways SafeUpdates now available.
- Flow-matching algorithms in JAX☆119Aug 12, 2024Updated 2 years ago
- Speech emotion recognition using LSTM, SVM and MLP | 语音情感识别☆10Jul 1, 2019Updated 7 years ago
- Mamba4Cast, a zero-shot time series forecasting model, achieves competitive performance and faster inference than transformer-based model…☆56Oct 11, 2024Updated 2 years ago
- Experiments on the impact of depth in transformers and SSMs.☆47Oct 23, 2025Updated 11 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 6 months ago
- ☆288Jun 6, 2025Updated last year