Crate burn_p2p_engine

Expand description

Internal engine integration helpers for burn-backed workloads.

Structs§

BurnArtifactBytes: Represents a burn artifact bytes.
BurnArtifactFile: Represents a burn artifact file.
BurnArtifactOptions: Represents a burn artifact options.
BurnEvaluator: Evaluates a model on a specific dataset.
BurnLearner: Learner struct encapsulating all components necessary to train a Neural Network model.
BurnLearningCheckpointer: Used to create, delete, or load checkpoints of the training process.
BurnMergeCandidate: Represents a burn merge candidate.
BurnModuleInventory: Represents a burn module inventory.
BurnModuleParameter: Represents a burn module parameter.
RecordArtifactFileOptions: Represents a record artifact file options.
StoreArtifactFileOptions: Represents a store artifact file options.

BurnRecordBytesFormat: Enumerates the supported burn record bytes format values.
BurnRecordFileFormat: Enumerates the supported burn record file format values.
BurnRecordPrecision: Enumerates the supported burn record precision values.
BurnStoreFormat: Enumerates the supported burn store format values.
BurnTensorKind: Enumerates the supported burn tensor kinds.
EngineError: Enumerates the supported engine error values.

apply_root_ema_modules: Performs the apply root EMA modules operation.
encode_record_bytes: Performs the encode record bytes operation.
encode_store_bytes: Performs the encode store bytes operation.
flatten_module_float_parameters: Exports deterministic flattened float parameters from a Burn module.
inspect_module: Performs the inspect module operation.
load_record_bytes: Performs the load record bytes operation.
load_record_file: Performs the load record file operation.
load_store_bytes: Performs the load store bytes operation.
load_store_file: Performs the load store file operation.
materialize_record_bytes_artifact: Performs the materialize record bytes artifact operation.
materialize_record_file_artifact: Performs the materialize record file artifact operation.
materialize_store_bytes_artifact: Performs the materialize store bytes artifact operation.
materialize_store_file_artifact: Performs the materialize store file artifact operation.
merge_weighted_mean_modules: Performs the merge weighted mean modules operation relative to a shared base module.
module_schema_hash: Performs the module schema hash operation.
replace_module_float_parameters: Restores a flattened float-parameter pack into a Burn module with the same layout.
save_record_file: Performs the save record file operation.
save_store_file: Performs the save store file operation.