orbax icon indicating copy to clipboard operation
orbax copied to clipboard
verified publisher icon google

Introduce `ListKey` and `TupleKey` for `list` and `tuple` PyTree nodes respectively.

Open copybara-service[bot] opened this issue 1 year ago • 0 comments

Introduce ListKey and TupleKey for list and tuple PyTree nodes respectively.

copybara-service[bot] avatar Oct 01 '24 22:10 copybara-service[bot]