Commits

Branch selector

 main

User selector

All users

Datepicker

All time

Commit History

Commits on Mar 1, 2024

• Fix github action version (missed in previous commit)

  Show description for c17e770
  romanngg

  committed
  c17e770

Commits on Feb 21, 2024

• Bump github actions versions to avoid test failures.

  Show description for 435be85
  romanngg

  committed
  435be85

Commits on Feb 20, 2024

• Avoid deprecated ad.config & ad.source_info_util

  Show description for e84d2ab
  

  Jake VanderPlas

  authored and

  romanngg

  committed
  e84d2ab

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.

  Show description for e91c80e
  romanngg

  committed
  e91c80e

• Remove now redundant mask type casting. Minor formatting fixes.

  Show description for 06c8ad8
  romanngg

  committed
  06c8ad8

• Remove tests for zeros_like and add_any primitives from neural_tangents.

  Show description for f96f176
  

  hawkinsp

  authored and

  romanngg

  committed
  f96f176

Commits on Dec 11, 2023

• Bump min JAX version to 0.4.16 + minor no-op linter fixes.

  Show description for ad47437
  romanngg

  committed
  ad47437

• Avoid use of deprecated device_buffer attriutes of jax.Array

  Show description for ad3d524
  

  Jake VanderPlas

  authored and

  romanngg

  committed
  ad3d524

Commits on Nov 21, 2023

• Support TF 2.15 (https://github.com/tensorflow/tensorflow/releases/tag/v2.15.0)

  Show description for 429dc13
  romanngg

  committed
  429dc13

• [JAX] Replace uses of jax.devices("cpu") with jax.local_devices(backend="cpu").

  Show description for aa3620d
  

  hawkinsp

  authored and

  romanngg

  committed
  aa3620d

• Resolve unsoundness caught by pytype --strict-none-binding.

  Show description for 3e10a78
  

  Felipe Wilches-Bernal

  authored and

  romanngg

  committed
  3e10a78

• neural_tangents: migrate from deprecated jax.linear_util to jax.extend.linear_util

  Show description for 8d948dc
  

  Jake VanderPlas

  authored and

  romanngg

  committed
  8d948dc

• Updated jax.config import

  Show description for 109796e
  

  superbobry

  authored and

  romanngg

  committed
  109796e

• Updated jax.config import

  Show description for dce935b
  

  superbobry

  authored and

  romanngg

  committed
  dce935b

• [LSC] change uses of jax.random.KeyArray and jax.random.PRNGKeyArray to jax.Array

  Show description for d816c8f
  

  Jake VanderPlas

  authored and

  romanngg

  committed
  d816c8f

Commits on Sep 30, 2023

• Minor fixes to make NT build at OSS head.

  Show description for 136338d
  romanngg

  committed
  136338d

• [LSC] Ignore incorrect type annotations related to jax.numpy APIs

  Show description for 271dcbe
  

  Jake VanderPlas

  authored and

  romanngg

  committed
  271dcbe

Commits on Sep 28, 2023

• Update branching.py

  romanngg

  authored
  09d7646

• Update branching.py

  romanngg

  authored
  4746cc2

• Update setup.py

  romanngg

  authored
  fc11f58

Commits on Sep 26, 2023

• Update empirical.py

  romanngg

  authored
  b170aaa

• Drop dependence on typing_extensions since typing Protocols are available in python >= 3.8

  Show description for cca7385
  romanngg

  committed
  cca7385

• Add missing copyright notice

  Show description for d839deb
  romanngg

  committed
  d839deb

• Migrate GenericFunction usages to PolymorphicFunction

  Show description for 8064529
  

  faizan-m

  authored and

  romanngg

  committed
  8064529

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.

  Show description for 3c3dc9f
  romanngg

  committed
  3c3dc9f

Commits on Aug 24, 2023

• Add python 3.11 support to pypi description.

  Show description for ae888ca
  romanngg

  committed
  ae888ca

• Fix crashing Github tests

  Show description for 91846d8
  romanngg

  committed
  91846d8

• Update readthedocs requirements to include all neural tangents build dependencies. Fix pytype python version

  Show description for b3b728e
  romanngg

  committed
  b3b728e

• Update predict_test.py

  romanngg

  authored
  db53e46

Commits on Aug 22, 2023

• Update linux.yml

  romanngg

  authored
  0f9bb34

• Update test_utils.py

  romanngg

  authored
  b8306f0

Commits on Aug 14, 2023

• Update pytype.yml

  romanngg

  authored
  5ffa3d3

• Update requirements.txt

  romanngg

  authored
  61c731e

• Revert to using Union instead of | for union types to support python 3.9 (| only works since python 3.10). Update flax dependency.

  Show description for 152d0c5
  romanngg

  committed
  152d0c5

• Minor linter / typos fixes

  Show description for e3e0f52
  romanngg

  committed
  e3e0f52

Pagination

PreviousNext