Separate function for inputs in fourier and spectral domain

[jakharkaran]

JAX (jit) does not accept logical operators, or lot of software engineering would be needed to do so.
So multiple functions are written for every possible flag of the function which will use jit or will be used in solver.

Examples
The name function takes inputs in spectral domain function_name_spectral
The name function takes inputs in physical domain function_name_physical