mfai.pytorch

mfai.pytorch.assign(left, right)

Used when loading weights coming from another training framework in to pytorch models. Checks the shapes matches and creates the learnable parameters from the supplied weights (rights). Copied from the llm from scratch repo “as-is”.

mfai.pytorch.export_to_onnx(model, sample, filepath, **kwargs)

Exports a model to ONNX format.

Return type:

None

Parameters:

• model (Module)

• sample (Tensor | tuple[Any, ...])

• filepath (Path | str)

• kwargs (Any)

mfai.pytorch.onnx_load_and_infer(filepath, input)

Loads a model using onnx, checks it, and performs an inference.

Return type:

ndarray

Parameters:

• filepath (Path | str)

• input (Tensor | tuple[Tensor, ...])

mfai.pytorch.to_numpy(input)

Return type:

ndarray | tuple[ndarray, ...]

Parameters:

input (Tensor | tuple[Tensor, ...])

• mfai.pytorch.lightning_modules
• mfai.pytorch.losses
• mfai.pytorch.models