You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Is your feature request related to a problem? Please describe.
RWKV and MAMBA are new families of models that are efficient. It would be great if TIMM lib have this
Describe the solution you'd like
add a new file in timm to implement them.
@2catycm they are both interesting model families, the problem is they all require custom kernels (or external libraries with custom kernels). Those have proven to be difficult to maintain over time, other models requiring custom kernels have come and gone in the time frame timm has existed and original authors did not maintain those kernels for current iterations of cuda/torch.
Is your feature request related to a problem? Please describe.
RWKV and MAMBA are new families of models that are efficient. It would be great if TIMM lib have this
Describe the solution you'd like
add a new file in timm to implement them.
Describe alternatives you've considered
Additional context
Here is the official lib of Vision-RWKV https://github.com/OpenGVLab/Vision-RWKV , and here is the paper https://openreview.net/forum?id=nGiGXLnKhl . And for RWKV this is the website https://rwkv.cn/ .
As for mamba, there are a lot of vision implementations. For example
With papers
The text was updated successfully, but these errors were encountered: