feat(mamba): add Mamba2 implementation

Add initial project structure including core Mamba2 logic, entry point, and uv-based dependency management.
2026-01-21 12:54:49 +08:00
commit c58a73ae26
8 changed files with 2151 additions and 0 deletions
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -0,0 +1,13 @@
+[project]
+name = "test-diffusion"
+version = "0.1.0"
+description = "Add your description here"
+readme = "README.md"
+requires-python = ">=3.12"
+dependencies = [
+    "einops>=0.7.0",
+    "matplotlib>=3.8.0",
+    "numpy>=1.26.0",
+    "swanlab>=0.5.0",
+    "torch>=2.2.0",
+]