v0.3.2

romanngg released this 27 Jul 13:51

Arbitrary nonlinearity approximation via Gaussian quadrature (stax.NumericalActivation)
Initial support for Graph Neural Networks (stax.Aggregate)
Further improvements and new features for attention (stax.GlobalSelfAtttention) from Infinite attention: NNGP and NTK for deep attention networks

Assets 4