mamba2-jax: A pure JAX/Flax implementation of Mamba-2 for language modeling and time series forecasting.
☆18Jun 23, 2026Updated last month
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:
- nanoGPT using Equinox☆15Mar 3, 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
- suffix array construction and searching algorithms for in-memory binary data.☆13Sep 10, 2022Updated 3 years ago
- Official implementation for the CVPR 2024 paper: HuProSO3: Normalizing Flows on the Product Space of SO(3) Manifolds for Probabilistic Hu…☆16Mar 31, 2025Updated last year
- ☆46Nov 6, 2025Updated 9 months ago
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click. Zero configuration with optimized deployments.
- Flax (Jax) implementation of DeepSeek-R1-Distill-Qwen-1.5B with weights ported from Hugging Face.☆26Feb 20, 2025Updated last year
- Xtructure is datastructure for using in JAX☆23Aug 2, 2026Updated last week
- Following research on S4 in jax☆16Jun 15, 2022Updated 4 years ago
- Differentiable and GPU accelerated scattering covariance statistics on the sphere☆16May 27, 2025Updated last year
- Convertible environment for reinforcement learning with Kerbal Space Program☆11Dec 8, 2022Updated 3 years ago
- [NAACL'25] Evaluating LLMs for Causal Queries☆14Feb 18, 2025Updated last year
- Concept Learning Dynamics☆17Oct 29, 2024Updated last year
- Implementation of physics informed neural networks with PyTorch☆22Jan 5, 2021Updated 5 years ago
- Code for the paper "BPE stays on SCRIPT", "Which Pieces Does Unigram Tokenization Really Need?" and MinGram☆19Updated 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.
- qwen3 experiments☆33Jul 1, 2025Updated last year
- Repository for "Self-Distillation for Model Stacking Unlocks Cross-Lingual NLU in 200+ Languages"☆15Oct 4, 2024Updated last year
- [ICML‘25] Official code for paper "Occult: Optimizing Collaborative Communication across Experts for Accelerated Parallel MoE Training an…☆13Apr 17, 2025Updated last year
- ☆18Apr 24, 2024Updated 2 years ago
- DIVA: A Dirichlet Process Mixtures Based Incremental Deep Clustering Algorithm via Variational Auto-Encoder☆13Sep 18, 2023Updated 2 years ago
- Archives for Triton Inference Server Practices☆15Feb 28, 2022Updated 4 years ago
- [ICML 2024] Code for the paper "MoE-RBench: Towards Building Reliable Language Models with Sparse Mixture-of-Experts"☆11Jul 1, 2024Updated 2 years ago
- Repository for Sparse Universal Transformers☆20Oct 23, 2023Updated 2 years ago
- Master thesis: Exploring bias in German NLG (GPT-3 & GerPT-2). Applies regard classification and bias mitigation triggers.☆16Sep 25, 2024Updated 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.
- Python script sending push notifications whenever a new room is published on the WOKO (student association for housing in Zurich) website…☆13Jul 13, 2023Updated 3 years ago
- See https://github.com/autogluon/tabarena for the newest exampels!☆15Nov 5, 2025Updated 9 months ago
- Official implementation of HART: Human Aligned Reconstruction Transformer☆17Oct 1, 2025Updated 10 months ago
- (SIGGRAPH Asia 2025) Pytorch implementation of "TrackerSplat: Exploiting Point Tracking for Fast and Robust Dynamic 3D Gaussians Reconst…☆19Apr 26, 2026Updated 3 months ago
- ☆18Apr 17, 2026Updated 3 months ago
- Support repository for the functional programming lecture (Haskell)☆19Jul 27, 2025Updated last year
- Differentiable and accelerated wavelet transform on the sphere with JAX☆26May 27, 2025Updated last year
- Code for the paper "Getting the most out of your tokenizer for pre-training and domain adaptation"☆22Feb 14, 2024Updated 2 years ago
- Developing hybrid deep learning models by integrating Neural networks with (s,e,t)GARCH models to predict volatility in the Indian Commod…☆19May 21, 2021Updated 5 years ago
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- An arbitrage bot is a smart contract connected to an external automation script that controls its operation.☆2,595Updated this week
- streaming deep reinforcement learning but 4x faster with jax!☆19Jan 4, 2026Updated 7 months ago
- ☆17Aug 20, 2025Updated 11 months ago
- Collection of small tips and tricks for C++☆18Mar 20, 2024Updated 2 years ago
- Vocal Tract Area Estimation by Gradient Descent☆39Jul 16, 2023Updated 3 years ago
- <알파제로를 분석하며 배우는 인공지능> 리포지토리☆14Feb 18, 2020Updated 6 years ago
- A collection of utilities for handling IPA phones.☆27Sep 24, 2023Updated 2 years ago