glassroom / torch_parallel_scanView on GitHub
Simple and fast parallel scan over sequences of tensors, with any binary associative function you specify, for PyTorch (Franz A. Heinsen, 2024)
18Apr 15, 2025Updated last year

Alternatives and similar repositories for torch_parallel_scan

Users that are interested in torch_parallel_scan are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.

Sorting:

Are these results useful?