proger / accelerated-scan

Accelerated First Order Parallel Associative Scan

https://twitter.com/darkproger/status/1745041586394648975

proger/accelerated-scan Issues

Torch 2.2 breakage on bfloat16 and float16
Updated 3 months ago
Is there any solution to support variable sequence length during training?
Updated 3 months ago1
Integrate with Mamba
Updated 5 months ago11
Support float16 and bfloat16
Closed 5 months ago