jax.experimental.serialize_executable.serialize#

jax.experimental.serialize_executable.serialize(compiled)[原始碼]#

序列化已編譯的二進位檔。

由於 pytree 無法序列化，因此會傳回它們，以便使用者可以正確處理。

參數:

compiled (jax.stages.Compiled)