Commit 1a1b71a

authored and

Orbax Authors

committed

Add PyTorchLayout for loading PyTorch checkpoints.

PiperOrigin-RevId: 821011541

1 parent 5585bf0 commit 1a1b71aCopy full SHA for 1a1b71a

3 files changed

+539

-0

lines changed

checkpoint
- orbax/checkpoint/experimental/v1/_src/layout
  - pytorch_layout.py
  - pytorch_layout_test.py
- pyproject.toml

+539

-0

lines changed

Comments

(0)