8000 Commits · lippirk/neural-tangents · GitHub

More Web Proxy on the site http://driver.im/

Commits

Commits on Mar 1, 2024

Fix github action version (missed in previous commit)
romanngg
committed

Commits on Feb 21, 2024

Bump github actions versions to avoid test failures.
romanngg
committed

Commits on Feb 20, 2024

Avoid deprecated ad.config & ad.source_info_util

Jake VanderPlas
authored and
romanngg
committed

Commits on Feb 1, 2024

Do not use dummy depthwise convolution on GPUs anymore, as performance on single-channel convolutions is now reasonable in latest CuDNN versions.
romanngg
committed
Remove now redundant mask type casting. Minor formatting fixes.
romanngg
committed
Remove tests for zeros_like and add_any primitives from neural_tangents.

hawkinsp
authored and
romanngg
committed

Commits on Dec 11, 2023

Bump min JAX version to 0.4.16 + minor no-op linter fixes.
romanngg
committed
Avoid use of deprecated `device_buffer` attriutes of jax.Array

Jake VanderPlas
authored and
romanngg
committed

Commits on Nov 21, 2023

Support TF 2.15 (https://github.com/tensorflow/tensorflow/releases/tag/v2.15.0)
romanngg
committed
[JAX] Replace uses of jax.devices("cpu") with jax.local_devices(backend="cpu").

hawkinsp
authored and
romanngg
committed
Resolve unsoundness caught by pytype --strict-none-binding.

Felipe Wilches-Bernal
authored and
romanngg
committed
neural_tangents: migrate from deprecated jax.linear_util to jax.extend.linear_util

Jake VanderPlas
authored and
romanngg
committed
Updated jax.config import

superbobry
authored and
romanngg
committed
Updated jax.config import

superbobry
authored and
romanngg
committed
[LSC] change uses of jax.random.KeyArray and jax.random.PRNGKeyArray to jax.Array

Jake VanderPlas
authored and
romanngg
committed

Commits on Sep 30, 2023

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

Jake VanderPlas
authored and
romanngg
committed

Commits on Sep 28, 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
committed

Commits on Aug 22, 2023

0