This repository was archived by the owner on May 6, 2025. It is now read-only.
Releases: google/neural-tangents
Releases · google/neural-tangents
v0.3.6
v0.3.5
New features:
- Major speedup of the empirical NTK via
vmap_axes- please see https://neural-tangents.readthedocs.io/en/latest/neural_tangents.empirical.html and discussion in #30 - Allow to compute maximum theoretical learning rate for a momentum optimizer
- Add an IMDB sentiment analysis example
- Allow pytrees as outputs of functions to linearize/taylorize
Breaking changes:
v0.3.4
v0.3.3
v0.3.2
v0.3.1
- New nonlinearities:
a Erf(b x + c),Gelu,RBF - Numpy
1.19compatibility - Bugfixes and documentation improvements