Official implementation of "Fourier Head: Helping Large Language Models Learn Complex Probability Distributions" - View it on GitHub