https://github.com/google/jax
History
Tip revision: e856ecfc95f79a6458ceb8b11ff6f40004baf50e authored by jax authors on 23 February 2022, 05:12:39 UTC
Introduce a new PjRtEvent class, and move BlockHostUntilReady and OnReady methods from PjRtBuffer into PjRtEvent, so instead of calling buf->BlockHostUntilReady() you now
Tip revision: e856ecf
File Mode Size
__init__.py -rw-r--r-- 756 bytes

back to top