Illustration dessinée par RepoMasters à partir de l’identifiant du dépôt et de la couleur de son langage. Crédits
PrototypeCommune

MzeroMiko/mamba-mini

An efficient pytorch implementation of selective scan in one file, works with both cpu and gpu, with corresponding mathematical derivation. It is probably the code which is the most close to selective_scan_cuda in mamba.

Propriétaire
MzeroMiko (compte personnel)
Étoiles
Étoiles : 108
Forks
Forks : 3
Langage
Python
Licence
Apache-2.0
Création
2024
Sujets
efficientmambapytorchselective-scan
Rang
Prototype (100 à 249 ★) Ce que veut dire ce rang
Voir sur GitHub

Autres dépôts de MzeroMiko

Les 3 dépôts de MzeroMiko

D’autres dépôts Python

Tous les dépôts Python