PyTorch-based implementations of short-time Fourier transform - View it on GitHub