For the complete documentation index, see llms.txt. Markdown versions of all pages are available by appending .md to any URL (e.g. /max/get-started.md).
Mojo module
selective_scan
comptime valuesβ
LOG2Eβ
comptime LOG2E = 1.4426950408889634
MAX_DSTATEβ
comptime MAX_DSTATE = 16
Strides1Dβ
comptime Strides1D = IndexList[1]
Strides2Dβ
comptime Strides2D = IndexList[2]
Strides3Dβ
comptime Strides3D = IndexList[3]
Strides4Dβ
comptime Strides4D = IndexList[4]
Functionsβ
- β
mamba_split_conv1d_scan_combined_cpu: CPU kernel for mamba_split_conv1d_scan_combined operation. - β
mamba_split_conv1d_scan_combined_gpu: GPU kernel for mamba_split_conv1d_scan_combined operation. - β
selective_scan_fwd_cpu: CPU kernel for selective scan forward pass. - β
selective_scan_fwd_cpu_minimal: Minimal CPU kernel for selective scan forward - no D, z, or delta_bias. - β
selective_scan_fwd_gpu: GPU kernel for selective scan forward pass. - β
selective_scan_fwd_gpu_minimal: Minimal GPU kernel for selective scan forward - no D, z, or delta_bias. - β
selective_scan_update_cpu: CPU kernel for selective scan update (single step). - β
selective_scan_update_gpu: GPU kernel for selective scan update (single step). - β
sigmoid: Optimized sigmoid using fast approximation for large negative values. - β
softplus: Softplus activation: log(1 + exp(x)) with numerical stability. - β
ssd_combined_cpu: CPU kernel for SSD combined operation. - β
ssd_combined_gpu: GPU kernel for SSD combined operation.
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!