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 3 months ago
- ☆23Nov 23, 2025Updated 7 months ago
- 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
- Partially Observable Multi-Agent RL with Transformers☆17Apr 27, 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.
- TiRex: Zero-Shot Forecasting Across Long and Short Horizons with Enhanced In-Context Learning☆287Jun 9, 2026Updated 3 weeks ago
- One Initialization to Rule them All: Fine-tuning via Explained Variance Adaptation☆53Oct 20, 2025Updated 8 months ago
- 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
- ☆19Nov 11, 2025Updated 7 months ago
- Machine learning algorithms implements with jax for machine learning in production in large scale dataset.☆19May 24, 2026Updated last month
- 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
- ☆10Nov 13, 2024Updated last year
- xLSTM as Generic Vision Backbone☆490Oct 20, 2025Updated 8 months ago
- Deploy open-source AI quickly and easily - Special Bonus Offer • AdRunpod Hub is built for open source. One-click deployment and autoscaling endpoints without provisioning your own infrastructure.
- ☆30Feb 27, 2024Updated 2 years ago
- A Julia package for real-time audification of ODEs and SDEs☆11Nov 24, 2025Updated 7 months ago
- SING: SDE Inference via Natural Gradients☆37May 5, 2026Updated last month
- [ICLR 2026] Official Implementation of ProxyThinker: Test-Time Guidance through Small Visual Reasoners.☆22Sep 24, 2025Updated 9 months ago
- Interactive timeline visualization for Org-roam.☆37Jan 12, 2026Updated 5 months ago
- Inkscape MCP Server - FastMCP-based server for Inkscape CLI and DOM operations with security boundaries and automatic parameter flattenin…☆44Oct 22, 2025Updated 8 months ago
- Code and data for paper "(How) do Language Models Track State?"☆25Mar 31, 2025Updated last year
- The implementation for FREE-Merging: Fourier Transform for Model Merging with Lightweight Experts (ICCV25)☆16Jun 26, 2025Updated last year
- Adapter-X: A Novel General Parameter-Efficient Fine-Tuning Framework for Vision☆11Jul 22, 2024Updated 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.
- ☆260Jun 6, 2025Updated last year
- Control inverted pendulum by LQR in OpenAI Gym☆12Oct 2, 2024Updated last year
- Reasoning Activation in LLMs via Small Model Transfer (NeurIPS 2025)☆22Oct 16, 2025Updated 8 months ago
- ☆27Nov 25, 2024Updated last year
- Implementation of Rational Kolmogorov-Arnold Network (rKAN)☆21Jun 28, 2024Updated 2 years ago
- ☆29Jan 8, 2026Updated 5 months ago
- ☆14Jan 16, 2025Updated last year
- ☆14Oct 21, 2024Updated last year
- Official repository for Activation-Informed Merging (AIM) of Large Language Models☆24Feb 10, 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.
- A simple script to see how my ideas evolve over time☆46Jun 4, 2025Updated last year
- Heatmap text in Julia.☆11Jul 3, 2025Updated last year
- HomebrewNLP in JAX flavour for maintable TPU-Training☆50Jan 20, 2024Updated 2 years ago
- Flow-matching algorithms in JAX☆119Aug 12, 2024Updated last year
- Pseudo-Marginal Slice Sampling☆19Jun 8, 2016Updated 10 years ago
- Efficient and Effective Weight-Ensembling Mixture of Experts for Multi-Task Model Merging. Arxiv, 2024.☆16Oct 28, 2024Updated last year
- Jax SSM Library☆48Nov 24, 2022Updated 3 years ago