Skip to content

Commit 4ac7f27

Browse files
author
Orbax Authors
committed
#p2p Add Grain data iterator checkpointing to P2P
PiperOrigin-RevId: 863828755
1 parent ac7c521 commit 4ac7f27

File tree

470 files changed

+968
-533
lines changed

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

470 files changed

+968
-533
lines changed

‎checkpoint/orbax/__init__.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
# Copyright 2025 The Orbax Authors.
1+
# Copyright 2026 The Orbax Authors.
22
#
33
# Licensed under the Apache License, Version 2.0 (the "License");
44
# you may not use this file except in compliance with the License.

‎checkpoint/orbax/checkpoint/__init__.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
# Copyright 2025 The Orbax Authors.
1+
# Copyright 2026 The Orbax Authors.
22
#
33
# Licensed under the Apache License, Version 2.0 (the "License");
44
# you may not use this file except in compliance with the License.

‎checkpoint/orbax/checkpoint/_src/arrays/abstract_arrays.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
# Copyright 2025 The Orbax Authors.
1+
# Copyright 2026 The Orbax Authors.
22
#
33
# Licensed under the Apache License, Version 2.0 (the "License");
44
# you may not use this file except in compliance with the License.

‎checkpoint/orbax/checkpoint/_src/arrays/abstract_arrays_test.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
# Copyright 2025 The Orbax Authors.
1+
# Copyright 2026 The Orbax Authors.
22
#
33
# Licensed under the Apache License, Version 2.0 (the "License");
44
# you may not use this file except in compliance with the License.

‎checkpoint/orbax/checkpoint/_src/arrays/fragments.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
# Copyright 2025 The Orbax Authors.
1+
# Copyright 2026 The Orbax Authors.
22
#
33
# Licensed under the Apache License, Version 2.0 (the "License");
44
# you may not use this file except in compliance with the License.

‎checkpoint/orbax/checkpoint/_src/arrays/fragments_test.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
# Copyright 2025 The Orbax Authors.
1+
# Copyright 2026 The Orbax Authors.
22
#
33
# Licensed under the Apache License, Version 2.0 (the "License");
44
# you may not use this file except in compliance with the License.

‎checkpoint/orbax/checkpoint/_src/arrays/numpy_utils.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
# Copyright 2025 The Orbax Authors.
1+
# Copyright 2026 The Orbax Authors.
22
#
33
# Licensed under the Apache License, Version 2.0 (the "License");
44
# you may not use this file except in compliance with the License.

‎checkpoint/orbax/checkpoint/_src/arrays/numpy_utils_test.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
# Copyright 2025 The Orbax Authors.
1+
# Copyright 2026 The Orbax Authors.
22
#
33
# Licensed under the Apache License, Version 2.0 (the "License");
44
# you may not use this file except in compliance with the License.

‎checkpoint/orbax/checkpoint/_src/arrays/sharding.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
# Copyright 2025 The Orbax Authors.
1+
# Copyright 2026 The Orbax Authors.
22
#
33
# Licensed under the Apache License, Version 2.0 (the "License");
44
# you may not use this file except in compliance with the License.

‎checkpoint/orbax/checkpoint/_src/arrays/sharding_test.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
# Copyright 2025 The Orbax Authors.
1+
# Copyright 2026 The Orbax Authors.
22
#
33
# Licensed under the Apache License, Version 2.0 (the "License");
44
# you may not use this file except in compliance with the License.

0 commit comments

Comments
 (0)