A novel implementation of fusing ViT with Mamba into a fast, agile, and high performance Multi-Modal Model. Powered by Zeta, the simplest AI framework ever.

Implementation of Vision Mamba from the paper: "Vision Mamba: Efficient Visual Representation Learning with Bidirectional State Space Model" It's 2.8x faster than DeiT and saves 86.8% GPU memory when performing batch inference to extract features on high-res images

Integrating Mamba/SSMs with Transformer for Enhanced Long Context and High-Quality Sequence Modeling

PyTorch Implementation of Jamba: "Jamba: A Hybrid Transformer-Mamba Language Model"

Minimal Mamba-2 implementation in PyTorch

Mambular is a Python package that brings the power of Mamba architectures to tabular data, offering a suite of deep learning models for regression, classification, and distributional regression tasks. This includes models like Mambular, FT-Transformer, TabTransformer and tabular ResNets.

A hierarchical yaml config in Python

An open source implementation of LFMs from Liquid AI: Liquid Foundation Models

[CVPR'24 Spotlight] The official implementation of "State Space Models for Event Cameras"

Implementation of PyTorch: "GAMBA: MARRY GAUSSIAN SPLATTING WITH MAMBA FOR SINGLE-VIEW 3D RECONSTRUCTION"

Official implementation of I2I-Mamba, an image-to-image translation model based on selective state spaces

Recall to Imagine, a model-based RL algorithm with superhuman memory. Oral (1.2%) @ ICLR 2024

A human-friendly way of managing parameters in AWS SSM

Some toy examples of score matching algorithms written in PyTorch

Neural State-Space Models and Latent Dynamics Functions in PyTorch for High-Dimensional Forecasting

Implementation of a modular, high-performance, and simplistic mamba for high-speed applications

Running the SSM agent as a Fargate task

A simpler Pytorch + Zeta Implementation of the paper: "SiMBA: Simplified Mamba-based Architecture for Vision and Multivariate Time series"

Structured Semantic Model supported Deep Neural Network for Click-Through Rate Prediction

[PRCV-2024] State Space Model based Frame-Event Tracking