Commits

Branch selector

 master

User selector

All users

Datepicker

All time

Commit History

Commits on Mar 29, 2023

• Remove references to deprecated jax.ShapedArray

  Show description for 1bb3b89
  

  Jake VanderPlas

  authored and

  copybara-github

  committed
  1bb3b89

  Copy full SHA for 1bb3b89

Commits on Feb 15, 2023

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

  Show description for e962caa
  

  hawkinsp

  authored and

  copybara-github

  committed
  e962caa

  Copy full SHA for e962caa

Commits on Dec 19, 2022

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

  Show description for a8fc2ce
  

  hawkinsp

  authored and

  copybara-github

  committed
  a8fc2ce

  Copy full SHA for a8fc2ce

Commits on Nov 9, 2022

• [tf.numpy] Fixes jax_tests/ breakages.

  Show description for 0b3ed2b
  

  wangpengmit

  authored and

  copybara-github

  committed
  0b3ed2b

  Copy full SHA for 0b3ed2b

Commits on Sep 9, 2022

• Make this code compatible with Python 3.10.

  Show description for 050e0b6
  

  yilei

  authored and

  copybara-github

  committed
  050e0b6

  Copy full SHA for 050e0b6

Commits on Aug 8, 2022

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

  Show description for 78d2b6e
  

  hawkinsp

  authored and

  copybara-github

  committed
  78d2b6e

  Copy full SHA for 78d2b6e

Commits on Aug 5, 2022

• A change to type checking in checkpoint loading.

  Show description for 99e9784
  

  henrykmichalewski

  authored and

  copybara-github

  committed
  99e9784

  Copy full SHA for 99e9784

Commits on Aug 4, 2022

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

  Show description for 03a1b45
  

  hawkinsp

  authored and

  copybara-github

  committed
  03a1b45

  Copy full SHA for 03a1b45

Commits on Jun 27, 2022

• Disable broken trax tests

  Show description for dfca406
  

  Jake VanderPlas

  authored and

  copybara-github

  committed
  dfca406

  Copy full SHA for dfca406

Commits on Jun 7, 2022

• Internal change

  Show description for e5eccdd
  

  trax-robot

  authored and

  copybara-github

  committed
  e5eccdd

  Copy full SHA for e5eccdd

Commits on May 17, 2022

• Internal change

  Show description for a29b0e2
  

  blakehechtman

  authored and

  copybara-github

  committed
  a29b0e2

  Copy full SHA for a29b0e2

Commits on Apr 24, 2022

• Remove unused comments related to Python 2 compatibility.

  Show description for 26c3455
  

  yilei

  authored and

  copybara-github

  committed
  26c3455

  Copy full SHA for 26c3455

Commits on Apr 22, 2022

• Remove unused comments related to Python 2 compatibility.

  Show description for 975725b
  

  yilei

  authored and

  copybara-github

  committed
  975725b

  Copy full SHA for 975725b

Commits on Apr 21, 2022

• Remove unused comments related to Python 2 compatibility.

  Show description for a0ac26d
  

  yilei

  authored and

  copybara-github

  committed
  a0ac26d

  Copy full SHA for a0ac26d

Commits on Apr 20, 2022

• Remove unused comments related to Python 2 compatibility.

  Show description for 41e0b27
  

  yilei

  authored and

  copybara-github

  committed
  41e0b27

  Copy full SHA for 41e0b27

• Remove unused comments related to Python 2 compatibility.

  Show description for 22c0906
  

  yilei

  authored and

  copybara-github

  committed
  22c0906

  Copy full SHA for 22c0906

• Remove unused comments related to Python 2 compatibility.

  Show description for deafd33
  

  yilei

  authored and

  copybara-github

  committed
  deafd33

  Copy full SHA for deafd33

Commits on Apr 15, 2022

• Remove unused comments related to Python 2 compatibility.

  Show description for e03d338
  

  yilei

  authored and

  copybara-github

  committed
  e03d338

  Copy full SHA for e03d338

Commits on Apr 14, 2022

• Remove dependence on private JAX test utilities

  Show description for 1010db3
  

  Jake VanderPlas

  authored and

  copybara-github

  committed
  1010db3

  Copy full SHA for 1010db3

Commits on Apr 13, 2022

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

  Show description for 12fc1e9
  

  hawkinsp

  authored and

  copybara-github

  committed
  12fc1e9

  Copy full SHA for 12fc1e9

Commits on Apr 4, 2022

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

  Show description for 918b1ce
  

  Jake VanderPlas

  authored and

  copybara-github

  committed
  918b1ce

  Copy full SHA for 918b1ce

Commits on Mar 22, 2022

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

  Show description for 83252f9
  

  hertschuh

  authored and

  copybara-github

  committed
  83252f9

  Copy full SHA for 83252f9

Commits on Feb 2, 2022

• Internal

  Show description for 31a2eaf
  

  Dmitry Tsarkov

  authored and

  copybara-github

  committed
  31a2eaf

  Copy full SHA for 31a2eaf

Commits on Dec 23, 2021

• Fix test failures caused by a NumPy 1.21 upgrade

  Show description for d6cae20
  

  hawkinsp

  authored and

  copybara-github

  committed
  d6cae20

  Copy full SHA for d6cae20

Commits on Dec 3, 2021

• Fix a failing initializer test.

  Show description for 51010fe
  

  henrykmichalewski

  authored and

  copybara-github

  committed
  51010fe

  Copy full SHA for 51010fe

Commits on Nov 16, 2021

• Fixes testTensordot flakiness by relaxing tolerance.

  Show description for 4c90f46
  

  wangpengmit

  authored and

  copybara-github

  committed
  4c90f46

  Copy full SHA for 4c90f46

Commits on Oct 27, 2021

• [JAX] Update users of jax.ops.index... functions, which are deprecated.

  Show description for c394f9d
  

  hawkinsp

  authored and

  copybara-github

  committed
  c394f9d

  Copy full SHA for c394f9d

Commits on Oct 26, 2021

• TRAX v.1.4.1 and updates to the release script for python3.

  Show description for 2f8f7e2
  

  afrozenator

  authored and

  copybara-github

  committed
  2f8f7e2

  Copy full SHA for 2f8f7e2

• Merge pull request google#1646 from OlegPonomaryov:batch_pad_fix

  Show description for 96cb76c
  copybara-github

  committed
  96cb76c

  Copy full SHA for 96cb76c

• TRAX v1.4.0

  Show description for de95e6e
  

  afrozenator

  authored and

  copybara-github

  committed
  de95e6e

  Copy full SHA for de95e6e

• The evaluation notebook for the NeurIPS paper "Sparse is Enough in Scaling Transformers".

  Show description for 2238490
  

  henrykmichalewski

  authored and

  copybara-github

  committed
  2238490

  Copy full SHA for 2238490

Commits on Oct 25, 2021

• Merge pull request google#1703 from sunvod:patch-1

  Show description for 283cbda
  copybara-github

  committed
  283cbda

  Copy full SHA for 283cbda

• Merge pull request google#1708 from sunvod:patch-2

  Show description for cf657f0
  copybara-github

  committed
  cf657f0

  Copy full SHA for cf657f0

• Merge pull request google#1707 from thoo:earlystopping_example

  Show description for 369958f
  copybara-github

  committed
  369958f

  Copy full SHA for 369958f

• Merge pull request google#1656 from manifest:init_checkpoint

  Show description for 8b04ec7
  copybara-github

  committed
  8b04ec7

  Copy full SHA for 8b04ec7

Pagination

Previous

Next