/fft-scan

Efficient PScan implementation in PyTorch

Primary LanguageJupyter Notebook