search

tancik / fourier-feature-networks

Fourier Features Let Networks Learn High Frequency Functions in Low Dimensional Domains
https://people.eecs.berkeley.edu/~bmild/fourfeat/
MIT License
1.27k stars 134 forks source link

fixed up import for jax changes #23

Closed johnrobinsn closed 1 year ago

johnrobinsn commented 1 year ago

Updated module imports for stax and optimizers.

johnrobinsn commented 1 year ago

Thanks for great notebook!