Commits

Branch selector

 master

User selector

All users

Datepicker

All time

Commit History

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

Commits on Oct 24, 2021

• Update rnn.py

  Show description for 7b86449
  sunvod

  authored
  7b86449

  Copy full SHA for 7b86449

Commits on Oct 21, 2021

• add a notebook for early stopping

  thoo

  committed
  cfed8d8

  Copy full SHA for cfed8d8

Commits on Oct 20, 2021

• Merge of PR google#1706

  Show description for b02b938
  

  syzymon

  authored and

  copybara-github

  committed
  b02b938

  Copy full SHA for b02b938

Commits on Oct 19, 2021

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

  Show description for 9a3c612
  

  hawkinsp

  authored and

  copybara-github

  committed
  9a3c612

  Copy full SHA for 9a3c612

• Merge pull request google#1700 from lukaszkaiser:master

  Show description for 4a81dc2
  copybara-github

  committed
  4a81dc2

  Copy full SHA for 4a81dc2

• Merge of PR google#1699

  Show description for 9dbe4be
  

  syzymon

  authored and

  copybara-github

  committed
  9dbe4be

  Copy full SHA for 9dbe4be

• [JAX] Replace users of deprecated jax.lib.xla_bridge functions with more public names.

  Show description for 3175ac0
  

  hawkinsp

  authored and

  copybara-github

  committed
  3175ac0

  Copy full SHA for 3175ac0

Commits on Oct 14, 2021

• Merge pull request google#1698 from Vatican-X-Formers:hourglass-pr

  Show description for 19739dd
  copybara-github

  committed
  19739dd

  Copy full SHA for 19739dd

Commits on Oct 8, 2021

• Update NMT_with_Transformers_Reformers_using_Trax.ipynb

  Show description for 2a1a0d2
  sunvod

  authored
  2a1a0d2

  Copy full SHA for 2a1a0d2

Commits on Oct 6, 2021

• Update gin config for enwik8

  syzymon

  committed
  be613ba

  Copy full SHA for be613ba

Commits on Oct 1, 2021

• [JAX] Replace uses of deprecated jax.ops.index_update(x, idx, y) APIs with their up-to-date, more succinct equivalent x.at[idx].set(y).

  Show description for f82d7a6
  

  hawkinsp

  authored and

  copybara-github

  committed
  f82d7a6

  Copy full SHA for f82d7a6

Commits on Sep 29, 2021

• Add depthwise convolution and allow to use it in CausalAttention.

  lukaszkaiser

  committed
  d875d80

  Copy full SHA for d875d80

Commits on Sep 16, 2021

• Make PrintShape work correctly when there is only one input

  syzymon

  committed
  5f7a27e

  Copy full SHA for 5f7a27e

• Fix duplication bug in generate_sequential_chunks when example is shorter than max_length

  syzymon

  committed
  2e1f6db

  Copy full SHA for 2e1f6db

• Update gin files for Hourglass

  syzymon

  committed
  3d2b954

  Copy full SHA for 3d2b954

• Add Hourglass

  syzymon

  committed
  1fb8dc3

  Copy full SHA for 1fb8dc3

Commits on Sep 11, 2021

• Merge pull request google#1696 from lukaszkaiser:lk

  Show description for 6151599
  copybara-github

  committed
  6151599

  Copy full SHA for 6151599

Commits on Sep 10, 2021

• Adding handling of GPU clusters and recent JAX renames.

  lukaszkaiser

  committed
  b6f5b06

  Copy full SHA for b6f5b06

• Make T5 imports dynamic so it can be optional.

  lukaszkaiser

  committed
  333172b

  Copy full SHA for 333172b

Commits on Aug 31, 2021

• Corrects link to TensorFlow NumPy in README.md

  wangpengmit

  authored
  a24ed37

  Copy full SHA for a24ed37

• Ensure values passed to jax.numpy functions are arrays rather than lists.

  Show description for 1389ceb
  

  Jake VanderPlas

  authored and

  copybara-github

  committed
  1389ceb

  Copy full SHA for 1389ceb

• [TRAX] Also install t5 while running tests.

  Show description for 9d9e0db
  

  afrozenator

  authored and

  copybara-github

  committed
  9d9e0db

  Copy full SHA for 9d9e0db

• Make installation of TRAX easier.

  Show description for 9b00295
  

  henrykmichalewski

  authored and

  copybara-github

  committed
  9b00295

  Copy full SHA for 9b00295

Commits on Aug 28, 2021

• A fix to a regression test.

  Show description for 2c08a94
  

  henrykmichalewski

  authored and

  copybara-github

  committed
  2c08a94

  Copy full SHA for 2c08a94

Commits on Aug 24, 2021

• Merge pull request google#1687 from Vatican-X-Formers:rotary-embeddings

  Show description for 25e0699
  copybara-github

  committed
  25e0699

  Copy full SHA for 25e0699

Pagination

Previous

Next