Code
Hub
Workspaces
Following
Trending
Connect
MCP
copy
Create free account
hub
/
github.com/LaurentMazare/mamba.rs
/ types & classes
Types & classes
16 in github.com/LaurentMazare/mamba.rs
⨍
Functions
60
◇
Types & classes
16
↓ 2 callers
Class
RMSNorm
jax/model.py:62
↓ 1 callers
Class
Config
jax/model.py:79
↓ 1 callers
Class
MambaBlock
jax/model.py:178
↓ 1 callers
Class
MambaModel
jax/model.py:250
↓ 1 callers
Class
ResidualBlock
jax/model.py:239
Class
Args
src/main.rs:56
Class
BlockWeights
src/model_inc.rs:10
Class
LinearNoBias
src/model.rs:7
Class
LogitsProcessor
src/main.rs:122
Class
MmapedWeights
This struct is self-referential in a sense as if mmap gets dropped, weights would not be valid anymore.
src/main.rs:12
Interface
ModelWeights
src/model.rs:85
Class
RmsNorm
src/model.rs:24
Class
State
src/model_inc.rs:34
Class
Tokenizer
src/tokenizer.rs:266
Class
Weights
src/model_inc.rs:27
Enum
Which
src/main.rs:41