boltzmann-Li

Results 1 issues of boltzmann-Li

All I did was moving the STFT computation outside the autocasting context, and added cast to float to make sure it's using float. Will also file an issue in pytorch...