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?