alxndrTL mamba.py issues

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 13 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