Implement scaling function calls as cuFFT callbacks

[mlxd]

This will remove unnecessary on-device memory transfers and host-kernel calls. Not sure of the expected performance increase, but it should be beneficial.