Made FFT run in parallel and broke out some helper functions.
Made FFT run in parallel and broke out some helper functions.

google-research/dex-lang,google-research/dex-lang,google-research/dex-lang,google-research/dex-lang,google-research/dex-lang,google-research/dex-lang