Commits

Branch selector

 main

User selector

All users

Datepicker

All time

Commit History

Commits on Sep 12, 2024

• Rename is_supported_empty_aggregation_type and is_supported_aggregation_type functions.

  

  niketkumar

  authored and

  t5-copybara

  committedSep 12, 2024
  705247b

Commits on Sep 3, 2024

• Remove external direct references to pytree_checkpoint_handler.

  

  cpgaffney1

  authored and

  t5-copybara

  committedSep 3, 2024
  c2f0c9c

• aggregate option cleanup.

  

  cpgaffney1

  authored and

  t5-copybara

  committedSep 3, 2024
  4d5716e

Commits on Aug 26, 2024

• Fix typo.

  

  agutkin

  authored and

  t5-copybara

  committedAug 26, 2024
  59a473c

Commits on Aug 20, 2024

• Split out a thin version of checkpoints.py that does not depend on JAX

  

  T5X Team

  authored and

  t5-copybara

  committedAug 20, 2024
  1909429

Commits on Aug 6, 2024

• Allow generating warmup data for any tokenizer type.

  

  T5X Team

  authored and

  t5-copybara

  committedAug 6, 2024
  0106fb6

Commits on Jul 30, 2024

• Fixing Issue google-research#1353

  

  Dnmh

  authored and

  t5-copybara

  committedJul 30, 2024
  ce68d6b

Commits on Jul 29, 2024

• Upgrade to python:3.10

  

  Dnmh

  authored and

  t5-copybara

  committedJul 29, 2024
  e1a6876

Commits on Jul 25, 2024

• [numpy] Fix users of NumPy APIs that are removed in NumPy 2.0.

  

  hawkinsp

  authored and

  t5-copybara

  committedJul 25, 2024
  f17e82f

• remove inlined jax.nn.initializers definitions, resolving TODO of levskaya et al

  

  mattjj

  authored and

  t5-copybara

  committedJul 25, 2024
  31410cc

Commits on Jul 16, 2024

• Enable ocdbt in t5x by default.

  

  liangyaning33

  authored and

  t5-copybara

  committedJul 16, 2024
  dcdcda9

Commits on Jul 15, 2024

• Stop writing msgpack file for new checkpoints and update empty nodes handling so that it no longer depends on this file.

  

  cpgaffney1

  authored and

  t5-copybara

  committedJul 15, 2024
  29ebb8e

Commits on Jul 8, 2024

• Stop writing msgpack file for new checkpoints and update empty nodes handling so that it no longer depends on this file.

  

  dubey

  authored and

  t5-copybara

  committedJul 8, 2024
  efce74c

Commits on Jul 3, 2024

• Stop writing msgpack file for new checkpoints and update empty nodes handling so that it no longer depends on this file.

  

  cpgaffney1

  authored and

  t5-copybara

  committedJul 3, 2024
  9c0cb7a

Commits on Jun 21, 2024

• Prevent divide by zero error in loss computation.

  

  T5X Team

  authored and

  t5-copybara

  committedJun 21, 2024
  f975a01

Commits on Jun 13, 2024

• Set aggregate=False under t5x.

  

  liangyaning33

  authored and

  t5-copybara

  committedJun 13, 2024
  645af9e

Commits on Jun 11, 2024

• Set aggregate to False under SaveArgs within t5x.

  

  liangyaning33

  authored and

  t5-copybara

  committedJun 11, 2024
  9e0093e

Commits on Jun 7, 2024

• Set aggregate to False under SaveArgs within t5x.

  

  liangyaning33

  authored and

  t5-copybara

  committedJun 7, 2024
  7f5ed21

Commits on May 24, 2024

• remove dependency on old protobuf

  

  iansimon

  authored and

  t5-copybara

  committedMay 24, 2024
  1aeb23e

• depend on orbax-checkpoint >= 0.5

  

  iansimon

  authored and

  t5-copybara

  committedMay 24, 2024
  09076a5

Commits on May 23, 2024

• Delegate to BasePyTreeCheckpointHandler rather than inheriting from it.

  

  cpgaffney1

  authored and

  t5-copybara

  committedMay 23, 2024
  cc77907

Commits on May 22, 2024

• Make OrbaxCheckpointManagerInterface gin.configurable.

  

  liangyaning33

  authored and

  t5-copybara

  committedMay 22, 2024
  7bd9794

Commits on May 21, 2024

• internal

  

  gauravmishra

  authored and

  t5-copybara

  committedMay 21, 2024
  3ea9b2c

Commits on May 17, 2024

• Fix sync_global_devices issue.

  

  liangyaning33

  authored and

  t5-copybara

  committedMay 17, 2024
  dddbfca

Commits on May 16, 2024

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

  

  Jake VanderPlas

  authored and

  t5-copybara

  committedMay 16, 2024
  a42d50b

Commits on May 15, 2024

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

  

  Jake VanderPlas

  authored and

  t5-copybara

  committedMay 15, 2024
  1938bf8

• Fix UnboundLocalError: local variable 'checkpoint_manager' referenced before assignment.

  

  liangyaning33

  authored and

  t5-copybara

  committedMay 15, 2024
  717cb3c

Commits on May 7, 2024

• Add ability to use checkpoints with non-standard directory name prefix.

  

  T5X Team

  authored and

  t5-copybara

  committedMay 7, 2024
  e84aab9

Commits on May 3, 2024

• Make CheckpointManagerConstructor gin-configurable.

  

  cpgaffney1

  authored and

  t5-copybara

  committedMay 3, 2024
  3843051

Commits on May 2, 2024

• change checkpointing to be compatible with new serialization

  

  T5X Team

  authored and

  t5-copybara

  committedMay 2, 2024
  6541a7c

Commits on Apr 30, 2024

• Use newer Orbax CheckpointManager API in T5X. This change is expected to be a no-op. Also removes a test case that is already covered in Orbax core.

  

  cpgaffney1

  authored and

  t5-copybara

  committedApr 30, 2024
  bd1fa1c

Commits on Apr 29, 2024

• Increase t5x AsyncCheckpointer timeout to 600s.

  

  liangyaning33

  authored and

  t5-copybara

  committedApr 29, 2024
  130910a

• Fixes typo in the documentation for predict_batch_with_aux.

  

  agutkin

  authored and

  t5-copybara

  committedApr 29, 2024
  e4e262b

Commits on Apr 22, 2024

• Enable AsyncCheckpointer by default in t5x.

  

  liangyaning33

  authored and

  t5-copybara

  committedApr 22, 2024
  6b02d25

Commits on Apr 18, 2024

• Fix typos in t5x/export_lib.py create_fake_input

  

  T5X Team

  authored and

  t5-copybara

  committedApr 18, 2024
  a81874c

Pagination

PreviousNext