Commits

Commits on Oct 12, 2023

Remove tensorflow
norabelrose
committedOct 12, 2023

Commits on Sep 30, 2023

Minor fixes to make NT build at OSS head.
romanngg
committedSep 30, 2023
[LSC] Ignore incorrect type annotations related to jax.numpy APIs

Jake VanderPlas
authored and
romanngg
committedSep 30, 2023

Commits on Sep 28, 2023

Update branching.py
romanngg
committedSep 28, 2023
Update branching.py
romanngg
committedSep 28, 2023
Update setup.py
romanngg
committedSep 28, 2023

Commits on Sep 26, 2023

Update empirical.py
romanngg
committedSep 26, 2023
Drop dependence on typing_extensions since typing Protocols are available in python >= 3.8
romanngg
committedSep 26, 2023
Add missing copyright notice
romanngg
committedSep 26, 2023
Migrate GenericFunction usages to PolymorphicFunction

faizan-m
authored and
romanngg
committedSep 26, 2023

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 24, 2023

Add python 3.11 support to pypi description.
romanngg
committedAug 24, 2023
Fix crashing Github tests
romanngg
committedAug 24, 2023
Update readthedocs requirements to include all neural tangents build dependencies. Fix pytype python version
romanngg
committedAug 24, 2023
Update predict_test.py
romanngg
committedAug 24, 2023

Commits on Aug 22, 2023

Update linux.yml
romanngg
committedAug 22, 2023
Update test_utils.py
romanngg
committedAug 22, 2023

Commits on Aug 14, 2023

Update pytype.yml
romanngg
committedAug 14, 2023
Update requirements.txt
romanngg
committedAug 14, 2023
Revert to using Union instead of | for union types to support python 3.9 (| only works since python 3.10). Update flax dependency.
romanngg
committedAug 14, 2023
Minor linter / typos fixes
romanngg
committedAug 14, 2023
Remove obsolete comment
romanngg
committedAug 14, 2023
Update readthedocs python version
romanngg
committedAug 14, 2023
Update github actions
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 Aug 11, 2023

Neural Tangents: Import of refs/pull/185/head
romanngg
committedAug 11, 2023
Fix pytype error
romanngg
committedAug 11, 2023

Commits on Aug 2, 2023

Remove references to deprecated jax aliases

Jake VanderPlas
authored and
jaehlee
committedAug 2, 2023
Deflake tests

romanngg
authored and
jaehlee
committedAug 2, 2023
Update to support JAX custom PRNG

Jake VanderPlas
authored and
jaehlee
committedAug 2, 2023

Commits on Jul 18, 2023

Correct arxiv links in README.md (google#185 )
mmhamdy
committedJul 18, 2023

Commits on May 8, 2023

Relax tests
romanngg
committedMay 8, 2023

Commits on Apr 19, 2023

Fix failing device_put rules tests
romanngg
committedApr 19, 2023
Do not use the potentially controversial qualifier "carpet" in README (see google#175 )
romanngg
committedApr 19, 2023
Migrate jax.ShapedArray -> jax.core.ShapedArray

Jake VanderPlas
authored and
romanngg
committedApr 19, 2023