mamba-rs 0.7.4

Mamba SSM and Mamba-3 SISO in Rust with optional CUDA acceleration: inference and training (BPTT through the SSM state, AdamW) on CPU and GPU, custom NVRTC-compiled kernels, CUDA Graph capture, f32 / bf16 / f16 storage, deterministic batch-invariant GEMMs by default with explicit cuBLAS Fast and Pedantic modes.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40

# The one gate. `make ci-fast` is what the pre-push hook runs and what
# CI runs: format, clippy with warnings as errors, the repository's own
# lints (xtask), and the dependency audit. `make check` adds the CPU
# test tier. GPU tiers run on a CUDA box (see docs/release-qualification.md), never here.
.PHONY: ci-fast check fmt fmt-check clippy lint deny test baseline install-hooks

ci-fast: fmt-check clippy lint deny

check: ci-fast test

fmt:
	cargo fmt --all

fmt-check:
	cargo fmt --all -- --check

clippy:
	cargo clippy --workspace --all-targets --locked -- -D warnings

# The repository's own rules (xtask): comments are plain English prose
# that says why - no Cyrillic, no task indexes, dates, document numbers
# or audit tags. A shrink-only baseline holds the existing stock; a new
# offending line fails here, in the hook and in CI alike.
lint:
	cargo run -q --locked -p xtask -- gate

deny:
	cargo deny check advisories bans sources

test:
	cargo test --workspace --locked

# Ratchet the comment baseline DOWN after rewriting comments. Refuses to
# grow: a new offending line is rewritten, never baselined.
baseline:
	cargo run -q -p xtask -- comments --update-baseline

install-hooks:
	install -m 0755 hooks/pre-push .git/hooks/pre-push
	@echo "pre-push installed: it runs make ci-fast before every push"