The current solution is to use tensors of shape [*, 2] where * is the shape of your complex array. All our spectral ops (*fft, stft) expect this format.
We are also working on enabling complex support. Actually the first couple PRs just landed on master.