issues
search
alxndrTL
/
mamba.py
A simple and efficient Mamba implementation in pure PyTorch and MLX.
MIT License
960
stars
86
forks
source link
issues
Newest
Newest
Most commented
Recently updated
Oldest
Least commented
Least recently updated
Question about the getdata() function
#59
peterphancong
opened
20 hours ago
0
The delta value is unreasonable during calculation on MacOS
#58
StepNeverStop
opened
4 days ago
1
pscan speed compared to simple for loop
#57
AnFreTh
closed
2 weeks ago
2
Question about converting mamba2 to onnx
#56
ToaTao
opened
3 weeks ago
1
Bibtex
#55
JulienSiems
closed
1 month ago
4
No module named 'mambapy.mamba_lm'
#54
ooooma
closed
1 month ago
1
Values of deltaA are very large
#53
anhtienng
closed
2 months ago
5
How to use cache in mamba2?
#52
wwwqqyy
closed
2 months ago
2
flops about mamba2
#51
dumpmemory
opened
2 months ago
2
muP for Mamba and Mamba-2
#50
alxndrTL
closed
2 months ago
4
MuP
#49
norikazu99
closed
2 months ago
4
[OLD] muP Mamba implementation & Mamba-2
#48
alxndrTL
closed
2 months ago
2
Up sweep in parallel scan
#46
anhtienng
closed
2 months ago
1
Can I translate your PScan in Jax?
#45
clementpoiret
closed
3 months ago
2
Default implementation of Jamba
#44
erlebach
closed
3 months ago
2
Partial batches in Mamba_lm
#43
erlebach
closed
3 months ago
1
Parallel pscan
#42
erlebach
closed
3 months ago
2
Error in the Mamba Block forward function?
#41
erlebach
closed
3 months ago
3
Question on using a sequence length > the max length I can hold in a batch due to memory usage for training
#40
luke-mcdermott-mi
opened
3 months ago
2
Pypi integration (pip install mambapy)
#39
alxndrTL
closed
3 months ago
0
support non-zero H[0] inputs
#38
Zizzzzzzz
closed
3 months ago
4
Cuda Version
#37
EddieEduardo
closed
4 months ago
5
Add mlx mamba training script
#36
michal1000w
closed
4 months ago
1
Pscan documentation
#35
poonam2308
opened
4 months ago
1
A basic implementation of ONNX exportation.
#34
Zeng-WCh
closed
3 months ago
1
<feat>: provide a naive onnx export
#33
Zeng-WCh
closed
5 months ago
1
Add Mamba language training script to pytorch examples folder
#32
michal1000w
closed
5 months ago
1
Fix MLX Generation script for Python 3.11
#31
michal1000w
closed
5 months ago
1
Add caches
#30
maciek-pioro
closed
5 months ago
0
VisionMamba implementation
#29
AliYoussef97
closed
5 months ago
6
[Feature Request] VideoMamba
#28
MelihDarcanxyz
opened
5 months ago
4
Changing th pscan from in-place to out-of-place?
#27
llmexperiment
closed
5 months ago
3
scan output is different between sequential and parallel versions
#26
llmexperiment
closed
5 months ago
3
delta question
#25
AliYoussef97
closed
5 months ago
5
Jamba PyTorch
#24
alxndrTL
closed
5 months ago
0
Please rectify paths in example_e2e_training.ipynb
#23
SiriusHou
closed
6 months ago
0
Working on AMD ROCm Platform
#22
supersonictw
closed
6 months ago
1
Update mamba_lm.py
#21
jaimepri
closed
5 months ago
1
A fresh can't start the model
#20
Luchen-077
closed
6 months ago
4
Integration to `transformers`
#19
ArthurZucker
closed
6 months ago
3
Possible SSM-Transformers implementation?
#18
severian42
opened
6 months ago
4
Why use element-wise multiplication rather than matrix multiplication in the function `selective_scan_seq`
#17
cszhbo
closed
6 months ago
2
Segmentation fault with MLX
#16
AshStill
closed
6 months ago
0
Support batch size > 1?
#15
yjdy
closed
6 months ago
1
Onnx export for the inference
#14
llmexperiment
closed
3 months ago
5
Mamba profiling_mamba.py script
#13
llmexperiment
closed
7 months ago
3
Performance update #1 : loop unfolding + reverse pscan for backward
#12
alxndrTL
closed
8 months ago
0
Can we get an explicit license?
#11
CompRhys
closed
8 months ago
1
Discretization step seems to be different from the paper
#10
ajtejankar
closed
8 months ago
3
huge huge memory usage!!
#9
eisneim
opened
8 months ago
8
Next