Commits

History for

Commits on Aug 29, 2023

No-op refactoring: use the modern jnp/np aliases convention for jax.numpy/numpy. Standardize hanging indentation in many-arguments functions. Fix minor typos / linter issues.
romanngg
committedAug 29, 2023

Commits on Aug 14, 2023

Remove obsolete comment
romanngg
committedAug 14, 2023
Use python >= 3.9 generic type annotations. Add support for 3.11, drop 3.8 following JAX / tf2jax. Upgrade dependencies and github actions.
romanngg
committedAug 14, 2023

Commits on Feb 11, 2023

Upgrade cuDNN version in Colabs to support JAX 0.4.3. Add missing "Open in Colab button".
romanngg
committedFeb 11, 2023

Commits on Dec 14, 2022

Multiple-descent learning curves for kernel regressions.

SiuMath
authored and
romanngg
committedDec 14, 2022

Commits on Sep 12, 2022

Add references to "Fast Neural Kernel Embeddings for General Activations"
romanngg
committedSep 12, 2022

Commits on Aug 19, 2022

Add new nonlinearities: Gabor, Polynomial, Monomial, RectifiedMonomial. Also make Hermite support arbitrary degrees.
romanngg
committedAug 19, 2022

Commits on Jun 25, 2022

Commits on Jun 24, 2022

Commits on Jun 20, 2022

Avoid TF version upgrade to prevent colab crashes.
romanngg
authoredJun 20, 2022

Commits on Jun 19, 2022

Add empirical NTK in Tensorflow colab example
romanngg
authoredJun 19, 2022
Fix name
romanngg
authoredJun 19, 2022
Point JAX installs in colabs to the new CUDA releases page.
romanngg
committedJun 19, 2022
Add an experimental prototype for using empirical NTK in Tensorflow.
romanngg
committedJun 19, 2022
Deflake a timing-out test
romanngg
committedJun 19, 2022

Commits on Jun 18, 2022

Add fast NTK examples
romanngg
committedJun 18, 2022

Commits on Jun 17, 2022

Remove some deprecated warnings/todos
romanngg
committedJun 17, 2022

Commits on Feb 17, 2022

Refactoring: a long awaited refactor that splits the huge stax.py into subcomponents, among other things:
romanngg
committedFeb 17, 2022

Commits on Feb 16, 2022

Automated g4 rollback of changelist 428678385.

Neural Tangents
authored and
romanngg
committedFeb 16, 2022
Refactoring: a long awaited refactor that splits the huge stax.py into subcomponents, among other things:
romanngg
committedFeb 16, 2022

Commits on Feb 14, 2022

Commits on Feb 8, 2022

Minor cleanup/refactoring:
romanngg
committedFeb 8, 2022

Commits on Nov 17, 2021

Following the release of JAX v0.2.25, rename jax.experimental to jax.example_libraries. Bump requirement to JAX v0.2.25 to avoid the CUDA-10 bug in google#122 (thanks @gortizji for pointing thi…
romanngg
committedNov 17, 2021

Commits on Oct 29, 2021

Temporarily revert jax-ml/jax@623c201 until JAX v0.2.25 is released, to make NT at head compatible with JAX at head and JAX at pypi.
romanngg
committedOct 29, 2021

Commits on Oct 28, 2021

Install neural tangents from pypi vs head in colabs. This helps avoid issues like google#127 , where NT head is fixed to work with JAX at head, but breaks when using the lagging colab JAX release.
romanngg
committedOct 28, 2021

Commits on Sep 18, 2021

Remove users of jax.api.* symbols, in preparation for removing the deprecated jax.api name.

hawkinsp
authored and
romanngg
committedSep 18, 2021

Commits on Jun 29, 2021

Replace references to master branches of NT/JAX to main. TF still uses master.
romanngg
committedJun 29, 2021

Commits on Jan 23, 2021

Squeeze tests more
romanngg
committedJan 23, 2021

Commits on Jan 21, 2021

Internal changes
romanngg
committedJan 21, 2021