mamba2-jax: A pure JAX/Flax implementation of Mamba-2 for language modeling and time series forecasting.
☆18Jun 23, 2026Updated 2 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:
- ☆11Dec 8, 2022Updated 3 years ago
- Ukrainian ELECTRA model☆12Mar 11, 2023Updated 3 years ago
- A Transfer Matrix Method Toolkit in Matlab☆12Oct 9, 2024Updated last year
- Piano Chords☆16May 9, 2026Updated 4 months ago
- A library for language transfer methods and algorithms.☆16Feb 6, 2026Updated 7 months ago
- AI Agents on 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.
- suffix array construction and searching algorithms for in-memory binary data.☆13Sep 10, 2022Updated 4 years ago
- Python module to remove wiki markup text.☆10Jan 15, 2016Updated 10 years ago
- Xtructure is datastructure for using in JAX☆23Updated this week
- Differentiable and GPU accelerated scattering covariance statistics on the sphere☆16May 27, 2025Updated last year
- Repository of "A Physics-Informed Neural Network Approach for Nearfield Acoustic Holography"☆19Apr 28, 2022Updated 4 years ago
- ☆18Mar 31, 2025Updated last year
- High quality implementations of imitation and inverse reinforcement learning algorithms☆25Aug 19, 2025Updated last year
- Fast reinforcement learning 💨☆29Jul 15, 2025Updated last year
- Implementation of physics informed neural networks with PyTorch☆23Jan 5, 2021Updated 5 years ago
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- Code for the paper "BPE stays on SCRIPT", "Which Pieces Does Unigram Tokenization Really Need?" and MinGram☆22Updated this week
- This is the official implement for LiveHPS++: Robust and Coherent Motion Capture in Dynamic Free Environment.☆17Sep 6, 2024Updated 2 years ago
- High-performance backend for language model probabilistic programs☆17Updated this week
- GCRL in JAX. Official repository for LEO (ICML 2026).☆33Jun 20, 2026Updated 2 months ago
- [ICML‘25] Official code for paper "Occult: Optimizing Collaborative Communication across Experts for Accelerated Parallel MoE Training an…☆14Apr 17, 2025Updated last year
- ☆18Apr 24, 2024Updated 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
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- Master thesis: Exploring bias in German NLG (GPT-3 & GerPT-2). Applies regard classification and bias mitigation triggers.☆16Sep 25, 2024Updated last year
- [NeurIPS 2022] Source code for our paper "Escaping Saddle Points for Effective Generalization on Class-Imbalanced Data"☆25Oct 16, 2023Updated 2 years ago
- (SIGGRAPH Asia 2025) Pytorch implementation of "TrackerSplat: Exploiting Point Tracking for Fast and Robust Dynamic 3D Gaussians Reconst…☆21Apr 26, 2026Updated 4 months ago
- Python monorepo template with Pants☆21Aug 19, 2023Updated 3 years ago
- Differentiable and accelerated wavelet transform on the sphere with JAX☆26May 27, 2025Updated last year
- Rust binding to crfsuite☆25Jan 31, 2026Updated 7 months ago
- Source code and data of our paper "Missing Counter-Evidence Renders NLP Fact-Checking Unrealistic for Misinformation" (https://arxiv.org/…☆11Jun 21, 2023Updated 3 years ago
- An extention to the GaLore paper, to perform Natural Gradient Descent in low rank subspace☆19Oct 21, 2024Updated last year
- An arbitrage bot is a smart contract connected to an external automation script that controls its operation.☆2,703Updated this week
- 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.
- ☆17Aug 20, 2025Updated last year
- Pytorch implementation of KFAC - this is a port of https://github.com/tensorflow/kfac/☆33Jun 6, 2024Updated 2 years ago
- A collection of utilities for handling IPA phones.☆27Sep 24, 2023Updated 2 years ago
- Multi-Turn RL Training System with AgentTrainer for Language Model Game Reinforcement Learning☆67Dec 18, 2025Updated 9 months ago
- Implementation of the SoftAdapt paper (techniques for adaptive loss balancing of multi-tasking neural networks)☆33Mar 16, 2024Updated 2 years ago
- Code for "The Geometry of Concepts: Sparse Autoencoder Feature Structure"☆18Mar 25, 2025Updated last year
- Generate a cute welcome message for yourself each day☆22Mar 30, 2023Updated 3 years ago