Serialization.SafeTensors Module Reference

Classes
class	Mila::Dnn::Serialization::SafeTensorsWriter
	Writes a safetensors file: u64 header length, JSON header, packed data. More...

Enumerations
enum class	Mila::Dnn::Serialization::TensorSavePass { Declare , Write }
	Which pass of a two-phase save a component is being driven through. More...

Functions
TensorDataType	Mila::Dnn::Serialization::fromSafeTensorsDataTypeName (const std::string &name)
	Mila type for a safetensors dtype spelling.
size_t	Mila::Dnn::Serialization::storageBytesPerElement (TensorDataType type)
	Byte width of one stored element of the given type.
std::string	Mila::Dnn::Serialization::toSafeTensorsDataTypeName (TensorDataType type)
	safetensors dtype spelling for a Mila type.

Variables
constexpr const char *	Mila::Dnn::Serialization::kMilaConfigMetadataKey = "mila_config"
	Key under which the Mila architecture config is stored in metadata.
constexpr const char *	Mila::Dnn::Serialization::kMilaQuantizationMetadataKey = "mila_quantization"
	Key marking an artifact whose weights are already quantized.

Files
file	Mila/Src/Dnn/Serialization/SafeTensors.ixx
	safetensors container support: dtype naming and a two-phase writer.