module
serialization
6 memberslucid.serializationlucid.serialization: save and load tensors and modules.
Functions
fn
save→ NoneSerialise `obj` to a Lucid-format checkpoint file.
fn
load→ objectLoad an object saved by `save` or `save_safetensors`.
fn
save_sharded→ NoneSave `obj` as a sharded, self-describing checkpoint directory.
fn
load_sharded→ OrderedDict or objectLoad a sharded checkpoint directory written by `save_sharded`.
fn
save_safetensors→ NoneSave a flat state dict as a SafeTensors file.
fn
load_safetensors→ dict of str to TensorLoad a SafeTensors file into a flat state dict.