Permalink
Failed to load latest commit information.
Type
Name
Latest commit message
Commit time
.github/workflows
Update readthedocs requirements to include all neural tangents build …
August 24, 2023 13:24
docs
Update requirements.txt
August 14, 2023 23:25
examples
No-op refactoring: use the modern jnp/np aliases convention for `…
August 29, 2023 10:04
neural_tangents
Support TF 2.15 (https://github.com/tensorflow/tensorflow/releases/ta…
November 21, 2023 03:40
notebooks
Updated jax.config import
November 21, 2023 03:39
presentation
Add paper
February 16, 2022 16:47
tests
[JAX] Replace uses of jax.devices("cpu") with jax.local_devices(backe…
November 21, 2023 03:40
.readthedocs.yml
Update readthedocs python version
August 14, 2023 12:34
CITATION
Deflake tests; update citation file
January 9, 2023 10:29
CONTRIBUTING.md
Project import generated by Copybara.
April 12, 2019 21:18
LICENSE
Refactoring: a long awaited refactor that splits the huge stax.py i…
February 16, 2022 19:27
LICENSE_SHORT
Add paper
February 16, 2022 16:47
README.md
No-op refactoring: use the modern jnp/np aliases convention for `…
August 29, 2023 10:04
setup.py
Support TF 2.15 (https://github.com/tensorflow/tensorflow/releases/ta…
November 21, 2023 03:40