Official JAX implementation of xLSTM including fast and efficient training and inference code. 7B model available at https://huggingface.co/NX-AI/xLSTM-7b.
☆105Jan 8, 2025Updated last year
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:
- ☆23Nov 23, 2025Updated 4 months ago
- Official repository of the xLSTM.☆2,141Nov 4, 2025Updated 5 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 10 months ago
- Partially Observable Multi-Agent RL with Transformers☆17Updated this week
- NordVPN Special Discount Offer • AdSave on top-rated NordVPN 1 or 2-year plans with secure browsing, privacy protection, and support for for all major platforms.
- TiRex: Zero-Shot Forecasting Across Long and Short Horizons with Enhanced In-Context Learning☆261Mar 23, 2026Updated 2 weeks ago
- ☆11Dec 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…☆49Oct 11, 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
- Emergent property inference.☆11Jul 6, 2023Updated 2 years ago
- ☆18Nov 11, 2025Updated 5 months ago
- Machine learning algorithms implements with jax for machine learning in production in large scale dataset.☆15Apr 1, 2026Updated last week
- 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.
- Optimised Extended LSTM for time-series forecasting☆41Nov 16, 2024Updated last year
- MLX implementation of xLSTM model by Beck et al. (2024)☆31Jun 5, 2024Updated last year
- xLSTM as Generic Vision Backbone☆490Oct 20, 2025Updated 5 months ago
- [ICLR 2026] Official Implementation of ProxyThinker: Test-Time Guidance through Small Visual Reasoners.☆21Sep 24, 2025Updated 6 months ago
- Resnets in tensorflow 2.0.☆17Mar 15, 2025Updated last year
- SING: SDE Inference via Natural Gradients☆36Dec 9, 2025Updated 4 months ago
- The implementation for FREE-Merging: Fourier Transform for Model Merging with Lightweight Experts (ICCV25)☆14Jun 26, 2025Updated 9 months ago
- Adapter-X: A Novel General Parameter-Efficient Fine-Tuning Framework for Vision☆11Jul 22, 2024Updated last year
- Python implementation of fast-oopsi, smc-oopsi, wiener, and discretize with binning☆15May 17, 2022Updated 3 years ago
- DigitalOcean Gradient AI Platform • AdBuild production-ready AI agents using customizable tools or access multiple LLMs through a single endpoint. Create custom knowledge bases or connect external data.
- Reasoning Activation in LLMs via Small Model Transfer (NeurIPS 2025)☆22Oct 16, 2025Updated 5 months ago
- Code for Research Project TLDR☆25Jul 28, 2025Updated 8 months ago
- ☆260Jun 6, 2025Updated 10 months ago
- Implementation of Rational Kolmogorov-Arnold Network (rKAN)☆21Jun 28, 2024Updated last year
- ☆38Jul 16, 2025Updated 8 months ago
- This is the open-source code for TokenCarve.☆26Jan 23, 2026Updated 2 months ago
- The official Languini Kitchen repository☆14May 6, 2024Updated last year
- Efficient encoder-decoder architecture for small language models (≤1B parameters) with cross-architecture knowledge distillation and visi…☆32Feb 7, 2025Updated last year
- ☆28Jan 8, 2026Updated 3 months 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.
- ☆13Jan 16, 2025Updated last year
- A simple script to see how my ideas evolve over time☆45Jun 4, 2025Updated 10 months ago
- ☆35Nov 22, 2024Updated last year
- Conformal Prediction for Time Series with Modern Hopfield Networks☆86May 28, 2025Updated 10 months ago
- Synthetic data generation for bangla OCR☆19Dec 1, 2022Updated 3 years ago
- the official code of "Diffusion-Based Image-to-Image Translation by Noise Correction via Prompt Interpolation" (ECCV2024)☆13Jan 14, 2025Updated last year
- NumPy+Jax with named axes and an uncompromising attitude☆23Mar 4, 2025Updated last year