GitHub repository: nate-gillman/fourier-head
Primary language: Python
Created: 2024-10-09T12:44:29Z
Description: Official implementation of "Fourier Head: Helping Large Language Models Learn Complex Probability Distributions" (ICLR 2025)