Looks like a good friend of mine, Prem Seetharaman, has written a pytorch implementation.
@smto What do you think about adding this to Pytorch?

  
      
      GitHub
  
  
    

pseeth/pytorch-stft 295

pytorch-stft - An STFT/iSTFT for PyTorch.