mamba2-jax: A pure JAX/Flax implementation of Mamba-2 for language modeling and time series forecasting.
☆19Jun 23, 2026Updated 4 months ago
Alternatives and similar repositories for mamba2-jax
Users that are interested in mamba2-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:
- JAX models for ML☆12Mar 2, 2026Updated 7 months ago
- A physics-informed neural network method for head-related transfer function upsampling☆10Feb 27, 2025Updated last year
- A Transfer Matrix Method Toolkit in Matlab☆11Oct 9, 2024Updated 2 years 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
- Code for Retrofitting Large Language Models with Dynamic Tokenization.☆14Jul 22, 2025Updated last year
- Managed Database hosting by DigitalOcean • AdPostgreSQL, MySQL, MongoDB, Kafka, Valkey, and OpenSearch available. Automatically scale up storage and focus on building your apps.
- ☆10Aug 26, 2025Updated last year
- Training neural networks to disentangle conservative and dissipative dynamics☆14Jan 27, 2022Updated 4 years ago
- Deep generative models for extending the bandwidth of reconstructed room impulse responses corrupted by undersampling.☆14Feb 14, 2023Updated 3 years ago
- Code for reproducing the paper "Neural Networks Fail to Learn Periodic Functions and How to Fix It" as part of the ML Reproducibility Cha…☆11Apr 16, 2021Updated 5 years ago
- NeMo: a toolkit for conversational AI☆13May 4, 2024Updated 2 years ago
- Toy meta-RL environments for testing algorithms implementations☆10Feb 21, 2024Updated 2 years ago
- A library for language transfer methods and algorithms.☆16Feb 6, 2026Updated 8 months ago
- Piano Chords☆16May 9, 2026Updated 5 months ago
- Convert DNA sequences to Midi: Listen to DNA☆11Mar 16, 2019Updated 7 years 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.
- suffix array construction and searching algorithms for in-memory binary data.☆13Sep 10, 2022Updated 4 years ago
- From Hero to Zéroe: A Benchmark of Low-Level Adversarial Attacks☆15Feb 23, 2023Updated 3 years ago
- JAX 문서 한글화 진행하는 프로젝트입니다.☆11Oct 18, 2023Updated 3 years ago
- ZMQ Connection Between Python and Unity☆10Jan 16, 2021Updated 5 years ago
- Xtructure is datastructure for using in JAX☆23Updated this week
- ☆51Nov 6, 2025Updated 11 months ago
- Continual Model Merging without Data: Dual Projections for Balancing Stability and Plasticity. NeurIPS, 2025.☆13Oct 12, 2025Updated last year
- Network representation learning on drug-target-side effects-indication graphs for side effect prediction☆13Feb 4, 2020Updated 6 years ago
- High quality implementations of imitation and inverse reinforcement learning algorithms☆26Aug 19, 2025Updated last year
- 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.
- Fast reinforcement learning 💨☆29Jul 15, 2025Updated last year
- Flax (Jax) implementation of DeepSeek-R1-Distill-Qwen-1.5B with weights ported from Hugging Face.☆26Feb 20, 2025Updated last year
- ☆19Mar 31, 2025Updated last year
- A python module implementing interfaces for various public tabular priors☆13Dec 28, 2025Updated 10 months ago
- Code and datasets for CVPR 2025 accepted paper Harnessing Frozen Unimodal Encoders for Flexible Multimodal Alignment☆13Jun 11, 2025Updated last year
- Community Collection of custom compile steps for the Longform Plugin for Obsidian☆19Feb 23, 2024Updated 2 years ago
- Fast wavelet transforms on the sphere.☆22Mar 30, 2025Updated last year
- Room Impulse Response Generator☆19May 20, 2018Updated 8 years ago
- JAX Version of VMEC2000☆16Updated this week
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- Repository of "A Physics-Informed Neural Network Approach for Nearfield Acoustic Holography"☆19Apr 28, 2022Updated 4 years ago
- ☆10Jul 14, 2018Updated 8 years ago
- The official code of TACL 2022, "Break, Perturb, Build: Automatic Perturbation of Reasoning Paths Through Question Decomposition".☆12Oct 18, 2021Updated 5 years ago
- Is Comples Query Answering Complex?☆14Oct 12, 2026Updated 2 weeks ago
- Repository for "Self-Distillation for Model Stacking Unlocks Cross-Lingual NLU in 200+ Languages"☆15Oct 4, 2024Updated 2 years ago
- Minimal, lightweight JAX implementations of popular models.☆246May 29, 2026Updated 5 months ago
- Official implementation for the CVPR 2024 paper: HuProSO3: Normalizing Flows on the Product Space of SO(3) Manifolds for Probabilistic Hu…☆18Mar 31, 2025Updated last year