AccFFT: A library for distributed-memory FFT on CPU and GPU architectures