Commits

Branch selector

 master

User selector

All users

Datepicker

All time

Commit History

Commits on Sep 10, 2024

• disable certain tests in lax_numpy_test when numpy 2.0 is used

  

  trax-robot

  authored and

  copybara-github

  committed
  8ca8408
  More actions

• disable testOp when numpy 2.0 is installed

  

  trax-robot

  authored and

  copybara-github

  committed
  6002f18
  More actions

• fix a few test bitwise op tests for np 2.0

  

  trax-robot

  authored and

  copybara-github

  committed
  17c19c7
  More actions

Commits on Aug 21, 2024

• skip testLinSpace and testLogSpace if numpy 2.0 is used

  

  trax-robot

  authored and

  copybara-github

  committed
  39c9bbb
  More actions

Commits on Aug 18, 2024

• [numpy] Fix test failures under NumPy 2.0.

  

  hawkinsp

  authored and

  copybara-github

  committed
  0d91963
  More actions

Commits on Aug 13, 2024

• [numpy] Fix test failures under NumPy 2.0.

  

  hawkinsp

  authored and

  copybara-github

  committed
  890052d
  More actions

Commits on Jul 16, 2024

• [scipy] Use NumPy functions rather than their deprecated aliases in SciPy's namespace.

  

  hawkinsp

  authored and

  copybara-github

  committed
  df631ba
  More actions

Commits on May 16, 2024

• Replace deprecated jax.tree_* functions with jax.tree.*

  

  Jake VanderPlas

  authored and

  copybara-github

  committed
  1451479
  More actions

Commits on Feb 1, 2024

• Remove uses of deprecated jax.lax.tie_in

  

  Jake VanderPlas

  authored and

  copybara-github

  committed
  a6a508e
  More actions

Commits on Nov 15, 2023

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

  

  hawkinsp

  authored and

  copybara-github

  committed
  d72bd65
  More actions

Commits on Oct 23, 2023

• trax: avoid referencing deprecated jax.random.threefry_2x32

  

  Jake VanderPlas

  authored and

  copybara-github

  committed
  38adb83
  More actions

Commits on Oct 20, 2023

• Updated jax.config import

  

  superbobry

  authored and

  copybara-github

  committed
  2a4356a
  More actions

Commits on Mar 29, 2023

• Remove references to deprecated jax.ShapedArray

  

  Jake VanderPlas

  authored and

  copybara-github

  committed
  1bb3b89
  More actions

Commits on Feb 15, 2023

• [JAX] Replace uses of jax.xla_computation() with jax.jit().lower().

  

  hawkinsp

  authored and

  copybara-github

  committed
  e962caa
  More actions

Commits on Dec 19, 2022

• [NumPy] Remove references to deprecated NumPy type aliases.

  

  hawkinsp

  authored and

  copybara-github

  committed
  a8fc2ce
  More actions

Commits on Nov 9, 2022

• [tf.numpy] Fixes jax_tests/ breakages.

  

  wangpengmit

  authored and

  copybara-github

  committed
  0b3ed2b
  More actions

Commits on Sep 9, 2022

• Make this code compatible with Python 3.10.

  

  yilei

  authored and

  copybara-github

  committed
  050e0b6
  More actions

Commits on Aug 8, 2022

• [NumPy] Replace numpy.asscalar(x) with x.item() in preparation for upgrading NumPy to 1.23.

  

  hawkinsp

  authored and

  copybara-github

  committed
  78d2b6e
  More actions

Commits on Aug 5, 2022

• A change to type checking in checkpoint loading.

  

  henrykmichalewski

  authored and

  copybara-github

  committed
  99e9784
  More actions

Commits on Aug 4, 2022

• [NumPy] Fix uses of deprecated multidimensional NumPy indexing with a non-tuple index.

  

  hawkinsp

  authored and

  copybara-github

  committed
  03a1b45
  More actions

Commits on Jun 27, 2022

• Disable broken trax tests

  

  Jake VanderPlas

  authored and

  copybara-github

  committed
  dfca406
  More actions

Commits on Jun 7, 2022

• Internal change

  

  trax-robot

  authored and

  copybara-github

  committed
  e5eccdd
  More actions

Commits on May 17, 2022

• Internal change

  

  blakehechtman

  authored and

  copybara-github

  committed
  a29b0e2
  More actions

Commits on Apr 24, 2022

• Remove unused comments related to Python 2 compatibility.

  

  yilei

  authored and

  copybara-github

  committed
  26c3455
  More actions

Commits on Apr 22, 2022

• Remove unused comments related to Python 2 compatibility.

  

  yilei

  authored and

  copybara-github

  committed
  975725b
  More actions

Commits on Apr 21, 2022

• Remove unused comments related to Python 2 compatibility.

  

  yilei

  authored and

  copybara-github

  committed
  a0ac26d
  More actions

Commits on Apr 20, 2022

• Remove unused comments related to Python 2 compatibility.

  

  yilei

  authored and

  copybara-github

  committed
  41e0b27
  More actions

• Remove unused comments related to Python 2 compatibility.

  

  yilei

  authored and

  copybara-github

  committed
  22c0906
  More actions

• Remove unused comments related to Python 2 compatibility.

  

  yilei

  authored and

  copybara-github

  committed
  deafd33
  More actions

Commits on Apr 15, 2022

• Remove unused comments related to Python 2 compatibility.

  

  yilei

  authored and

  copybara-github

  committed
  e03d338
  More actions

Commits on Apr 14, 2022

• Remove dependence on private JAX test utilities

  

  Jake VanderPlas

  authored and

  copybara-github

  committed
  1010db3
  More actions

Commits on Apr 13, 2022

• [JAX] Remove unused test utilities that access JAX internals that are about to be removed.

  

  hawkinsp

  authored and

  copybara-github

  committed
  12fc1e9
  More actions

Commits on Apr 4, 2022

• Fix issues related to new behavior of JAX DeviceArray.copy()

  

  Jake VanderPlas

  authored and

  copybara-github

  committed
  918b1ce
  More actions

Commits on Mar 22, 2022

• Explicitly import estimator from tensorflow as a separate import instead of

  

  hertschuh

  authored and

  copybara-github

  committed
  83252f9
  More actions

Commits on Feb 2, 2022

• Internal

  

  Dmitry Tsarkov

  authored and

  copybara-github

  committed
  31a2eaf
  More actions

Pagination

PreviousNext