nvmath-python brings the power of the NVIDIA math libraries to the Python ... epilogs for FFT functions as a Python functions and compiled them LTO-IR. For example, to have unitary scaling for an FFT, ...