Only implemented through torch: "bi - mamba2" , "vision- mamba2 -torch". support 1d/2d/3d/nd and support export by jit.script/onnx;
-
Updated
Dec 11, 2024 - Python
#
torch-script
Here is 1 public repository matching this topic...
Improve this page
Add a description, image, and links to the torch-script topic page so that developers can more easily learn about it.
Add this topic to your repo
To associate your repository with the torch-script topic, visit your repo's landing page and select "manage topics."