https://github.com/google/jax
History
Tip revision: d73319c8a25596af21dab9a9c12a468e1f6242d3 authored by Colin Gaffney on 10 June 2022, 22:09:01 UTC
Introduce AsyncCheckpointManager, which allows commit operations to take place in a background thread while training may continue in the main thread. This class inherits from both CheckpointManager, and reuses much of its functionality, and AsyncWriteManager in jax.experimental, which provides needed async utilities.
Tip revision: d73319c
File Mode Size
images
JAX_NeurIPS_2020_demo.ipynb -rw-r--r-- 10.5 KB
JAX_demo.ipynb -rw-r--r-- 19.5 KB
Lorentz_ODE_Solver.ipynb -rw-r--r-- 14.7 KB
Pmap_Cookbook.ipynb -rw-r--r-- 17.9 KB
README.md -rw-r--r-- 4.8 KB
Wave_Equation.ipynb -rw-r--r-- 15.5 KB

README.md

back to top