jax.experimental.serialize_executable 模組

預先編譯二進位檔的 Pickling 支援。

API

serialize(compiled)	序列化已編譯的二進位檔。
deserialize_and_load(serialized, in_tree, ...)	從序列化的可執行檔建構 jax.stages.Compiled。