Commits

Branch selector

 main

User selector

All users

Datepicker

All time

Commit History

Commits on Jun 30, 2025

• add %%capture to hide cell output in training example.

  Show description for 705778d
  

  proximityy

  authored and

  Orbax Authors

  committed
  705778d

  Copy full SHA for 705778d

• Make model save accept a visibility argument.

  Show description for 1791d1e
  

  Orbax Authors

  committed
  1791d1e

  Copy full SHA for 1791d1e

• Expose Preservation Policy APIs to user.

  Show description for 2309437
  

  abhishek002002

  authored and

  Orbax Authors

  committed
  2309437

  Copy full SHA for 2309437

• #emergency Eliminate usage of `slice_index` device property.

  Show description for e0f7351
  

  cpgaffney1

  authored and

  Orbax Authors

  committed
  e0f7351

  Copy full SHA for e0f7351

Commits on Jun 27, 2025

• Set default `tpu_compilation_environment` to xla `compile_options`.

  Show description for e85f11d
  

  jerryxyj

  authored and

  Orbax Authors

  committed
  e85f11d

  Copy full SHA for e85f11d

• No public description

  Show description for 39bdd44
  

  Orbax Authors

  committed
  39bdd44

  Copy full SHA for 39bdd44

• add opencv to pyproject.toml

  Show description for 84032ac
  

  proximityy

  authored and

  Orbax Authors

  committed
  84032ac

  Copy full SHA for 84032ac

• Place --xla_force_host_platform_device_count flag earlier in our example.

  Show description for c2da7b0
  

  proximityy

  authored and

  Orbax Authors

  committed
  c2da7b0

  Copy full SHA for c2da7b0

Commits on Jun 26, 2025

• #v1 Add PreservationPolicy support in Checkpointer.

  Show description for 20f546e
  

  cpgaffney1

  authored and

  Orbax Authors

  committed
  20f546e

  Copy full SHA for 20f546e

• Add tensorflow_datasets to pyproject.toml

  Show description for 9a1500c
  

  proximityy

  authored and

  Orbax Authors

  committed
  9a1500c

  Copy full SHA for 9a1500c

• No public description

  Show description for 4702c14
  

  Orbax Authors

  committed
  4702c14

  Copy full SHA for 4702c14

Commits on Jun 25, 2025

• V1: Remove unnecessary type handler add warnings

  Show description for f95f1fc
  

  ChromeHearts

  authored and

  Orbax Authors

  committed
  f95f1fc

  Copy full SHA for f95f1fc

• Revise MNIST training data example to include optimizer checkpointing and restoring a model state given a specific checkpoint path.

  Show description for b7afccf
  

  proximityy

  authored and

  Orbax Authors

  committed
  b7afccf

  Copy full SHA for b7afccf

• Internal change.

  Show description for 4ca2ab2
  

  cpgaffney1

  authored and

  Orbax Authors

  committed
  4ca2ab2

  Copy full SHA for 4ca2ab2

Commits on Jun 24, 2025

• Update sphinxcontrib-bibtex to latest version to address readthedocs build issue.

  Show description for f061e4a
  

  BlaziusMaximus

  authored and

  Orbax Authors

  committed
  f061e4a

  Copy full SHA for f061e4a

• Remove old heartbeat flags and arguments.

  Show description for 30d6ee6
  

  mwhittaker

  authored and

  Orbax Authors

  committed
  30d6ee6

  Copy full SHA for 30d6ee6

Commits on Jun 23, 2025

• In `SingleReplicaArrayHandler`, pass additional relevant constructor options through to the parent class.

  Show description for b87a223
  

  cpgaffney1

  authored and

  Orbax Authors

  committed
  b87a223

  Copy full SHA for b87a223

• No public description

  Show description for 0732084
  

  yashk2810

  authored and

  Orbax Authors

  committed
  0732084

  Copy full SHA for 0732084

Commits on Jun 22, 2025

• Use `Layout` in place of `DeviceLocalLayout` and `.layout` in place of `.device_local_layout`.

  Show description for 08c7b6e
  

  yashk2810

  authored and

  Orbax Authors

  committed
  08c7b6e

  Copy full SHA for 08c7b6e

• Use `Layout` in place of `DeviceLocalLayout`

  Show description for 4b887c7
  

  yashk2810

  authored and

  Orbax Authors

  committed
  4b887c7

  Copy full SHA for 4b887c7

Commits on Jun 21, 2025

• Add grain as a dependency in pyproject.toml

  Show description for f83e01b
  

  proximityy

  authored and

  Orbax Authors

  committed
  f83e01b

  Copy full SHA for f83e01b

Commits on Jun 20, 2025

• Add MNIST data example to training.ipynb.

  Show description for 91d3ce9
  

  proximityy

  authored and

  Orbax Authors

  committed
  91d3ce9

  Copy full SHA for 91d3ce9

• Update pytree_handler to use the v0/v1 compatibility type handler registry.

  Show description for b28a8e0
  

  ChromeHearts

  authored and

  Orbax Authors

  committed
  b28a8e0

  Copy full SHA for b28a8e0

• #v1 Add back `nest_asyncio.apply`.

  Show description for c868e09
  

  cpgaffney1

  authored and

  Orbax Authors

  committed
  c868e09

  Copy full SHA for c868e09

• No public description

  Show description for a34b2e4
  

  yashk2810

  authored and

  Orbax Authors

  committed
  a34b2e4

  Copy full SHA for a34b2e4

Commits on Jun 19, 2025

• Use .format in place of .layout.

  Show description for ab86f5a
  

  yashk2810

  authored and

  Orbax Authors

  committed
  ab86f5a

  Copy full SHA for ab86f5a

• Add `Function::data_names` to specify the data to load

  Show description for c83296e
  

  AspirinSJL

  authored and

  Orbax Authors

  committed
  c83296e

  Copy full SHA for c83296e

Commits on Jun 18, 2025

• Use `jax_layout.Format` in place of `jax_layout.Layout`

  Show description for bdb4597
  

  yashk2810

  authored and

  Orbax Authors

  committed
  bdb4597

  Copy full SHA for bdb4597

• No public description

  Show description for fc5eaad
  

  ChromeHearts

  authored and

  Orbax Authors

  committed
  fc5eaad

  Copy full SHA for fc5eaad

• Avoid using the default key in obm_export.py.

  Show description for 2b191f6
  

  jerryxyj

  authored and

  Orbax Authors

  committed
  2b191f6

  Copy full SHA for 2b191f6

• No public description

  Show description for 03f6234
  

  stjerngren

  authored and

  Orbax Authors

  committed
  03f6234

  Copy full SHA for 03f6234

Commits on Jun 17, 2025

• No public description

  Show description for d73ce4d
  

  cpgaffney1

  authored and

  Orbax Authors

  committed
  d73ce4d

  Copy full SHA for d73ce4d

• Reflect https://github.com/google/orbax/pull/2022.

  Show description for 6deb2f5
  

  cpgaffney1

  authored and

  Orbax Authors

  committed
  6deb2f5

  Copy full SHA for 6deb2f5

• internal monitoring and hashing change

  Show description for 1b768ac
  

  Orbax Authors

  committed
  1b768ac

  Copy full SHA for 1b768ac

• Merge pull request #2022 from keshavb96/correct_gpu_names

  Show description for 909e5d1
  cpgaffney1

  authored
  909e5d1

  Copy full SHA for 909e5d1

Pagination

Previous

Next