orbax
orbax copied to clipboard
Add a new parameter `background_thread_count` to allow users to set the number of threads for ThreadedCheckpointDeleter.
Add a new parameter background_thread_count to allow users to set the number of threads for ThreadedCheckpointDeleter.