search

google / orbax

Orbax provides common checkpointing and persistence utilities for JAX users
https://orbax.readthedocs.io/
Apache License 2.0
303 stars 36 forks source link

Add MultiProcessTest base. #1328

Open copybara-service[bot] opened 1 week ago

copybara-service[bot] commented 1 week ago

Add MultiProcessTest base.