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 4 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.☆94Jul 6, 2026Updated last month
- ☆22Aug 19, 2026Updated 2 weeks ago
- Official repository of the xLSTM.☆2,196Updated this week
- Dreamer on JAX☆16Jan 19, 2022Updated 4 years ago
- TiRex: Zero-Shot Forecasting Across Long and Short Horizons with Enhanced In-Context Learning☆299Updated this week
- 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 …☆30May 17, 2025Updated last year
- ☆12Dec 4, 2024Updated last year
- Implementation of our NeurIPS 2019 paper: Subspace Attack: Exploiting Promising Subspaces for Query-Efficient Black-box Attacks☆10Dec 16, 2019Updated 6 years ago
- [ICCV 2025] Code for "Flow Stochastic Segmentation Networks"☆18Jun 9, 2026Updated 2 months ago
- MLX implementation of xLSTM model by Beck et al. (2024)☆31Jun 5, 2024Updated 2 years ago
- Launching and monitoring Slurm experiments in Python☆34Jun 29, 2026Updated 2 months ago
- Clock In/Clock Out buttons per org heading to simplify clocking in whether on a phone or a desktop.☆11Feb 19, 2023Updated 3 years ago
- Resnets in tensorflow 2.0.☆17Mar 15, 2025Updated last year
- Proton VPN Special Offer - Get 70% off • AdSpecial partner offer. Trusted by over 100 million users worldwide. Tested, Approved and Recommended by Experts.
- One Initialization to Rule them All: Fine-tuning via Explained Variance Adaptation☆52Oct 20, 2025Updated 10 months ago
- ☆10Jan 12, 2024Updated 2 years ago
- ☆30Feb 27, 2024Updated 2 years ago
- ☆15Jan 16, 2025Updated last year
- Python implementation of fast-oopsi, smc-oopsi, wiener, and discretize with binning☆15May 17, 2022Updated 4 years ago
- ☆260Jun 6, 2025Updated last year
- The official Languini Kitchen repository☆14May 6, 2024Updated 2 years ago
- ☆29Jan 8, 2026Updated 7 months ago
- Implementation of Rational Kolmogorov-Arnold Network (rKAN)☆21Jun 28, 2024Updated 2 years ago
- Open source password manager - Proton Pass • AdSecurely store, share, and autofill your credentials with Proton Pass, the end-to-end encrypted password manager trusted by millions.
- ☆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
- Conformal Prediction for Time Series with Modern Hopfield Networks☆85May 28, 2025Updated last year
- Julia implementation of flash-attention operation for neural networks.☆11May 31, 2023Updated 3 years ago
- Heatmap text in Julia.☆11Jul 13, 2026Updated last month
- PyTorch Implementation on Paper [ICLR2026]: Foundation Models for Causal Inference via Prior-Data Fitted Networks☆51Jun 7, 2026Updated 2 months ago
- HomebrewNLP in JAX flavour for maintable TPU-Training☆50Jan 20, 2024Updated 2 years ago
- An emacs client for Bluesky☆19Jun 18, 2026Updated 2 months ago
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- Einsum-like high-level array sharding API for JAX☆35Jul 16, 2024Updated 2 years ago
- Machine learning algorithms implements with jax for machine learning in production in large scale dataset.☆21Aug 13, 2026Updated 2 weeks ago
- Official repository for Parallax (Parameterized Local Linear Attention)☆68Jul 30, 2026Updated last month
- official repository for the NeurIPS 2022 paper "Adversarial Attack on Attackers: Post-Process to Mitigate Black-Box Score-Based Query Att…☆20Oct 28, 2022Updated 3 years ago
- Official codebase for our paper "Do Language Models Use Their Depth Efficiently?"☆29Jun 25, 2025Updated last year
- Flow-matching algorithms in JAX☆119Aug 12, 2024Updated 2 years ago
- ☆19Nov 11, 2025Updated 9 months ago