Add fftfreq, rfftfreq and scalar axes for fftshift/ifftshift#3298

declanhealy2 wants to merge 3 commits intoml-explore:mainml-explore/mlx:mainfrom

declanhealy2:fftfreq-and-scalar-axesdeclanhealy2/mlx:fftfreq-and-scalar-axesCopy head branch name to clipboard