JAX > Fourier Feature Networks

Official implementation of Fourier Features Let Networks Learn High Frequency Functions in Low Dimensional Domains.