Skip to content

Conversation

@stefanradev93
Copy link
Contributor

Wraps Mamba State Space Model (SSM) as a powerful summary network for time series. Usage:

summary_net = MambaSSM()

Credit to: https://github.com/state-spaces/mamba/tree/main

Note: Only supports torch backend due to hardware-aware optimization.

@stefanradev93 stefanradev93 self-assigned this Apr 10, 2025
@stefanradev93 stefanradev93 added the feature New feature or request label Apr 10, 2025
@stefanradev93 stefanradev93 added this to the BayesFlow 2.0 milestone Apr 10, 2025
@stefanradev93 stefanradev93 requested a review from LarsKue April 10, 2025 01:05
@codecov
Copy link

codecov bot commented Apr 10, 2025

Codecov Report

Attention: Patch coverage is 59.42029% with 28 lines in your changes missing coverage. Please review.

Files with missing lines Patch % Lines
bayesflow/wrappers/mamba/mamba.py 57.57% 14 Missing ⚠️
bayesflow/wrappers/mamba/mamba_block.py 58.82% 14 Missing ⚠️
Files with missing lines Coverage Δ
bayesflow/wrappers/__init__.py 100.00% <100.00%> (ø)
bayesflow/wrappers/mamba/__init__.py 100.00% <100.00%> (ø)
bayesflow/wrappers/mamba/mamba.py 57.57% <57.57%> (ø)
bayesflow/wrappers/mamba/mamba_block.py 58.82% <58.82%> (ø)

... and 11 files with indirect coverage changes

Copy link
Contributor

@LarsKue LarsKue left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Looks good overall. I would change some minor things, see individual comments.

@LarsKue LarsKue merged commit 62cf5b5 into dev Apr 11, 2025
2 of 14 checks passed
@LarsKue LarsKue deleted the mamba branch April 11, 2025 15:41
@github-project-automation github-project-automation bot moved this from Future to Done in bayesflow development Apr 11, 2025
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

feature New feature or request

Projects

Status: Done

Development

Successfully merging this pull request may close these issues.

3 participants