mlx_io (mlx v0.2.0)

View Source

Summary

Functions

Functions

cleanup_checkpoints(CheckpointDir, KeepLast)

close_model_stream(StreamState)

compress_for_storage(Model, CompressionLevel)

compress_for_storage(Model, CompressionLevel, Options)

convert_format(InputPath, OutputPath, TargetFormat)

convert_format(InputPath, OutputPath, TargetFormat, Options)

convert_from_huggingface(ModelName, CacheDir)

create_model_stream(FilePath, Mode)

decompress_from_storage(CompressedModel)

export_onnx(Model, FilePath)

export_onnx(Model, FilePath, Options)

export_tensorflow(Model, FilePath)

export_tensorflow(Model, FilePath, Options)

extract_model_info(Model)

import_onnx(FilePath)

import_onnx(FilePath, Options)

import_tensorflow(SavedModelPath)

import_tensorflow(SavedModelPath, Options)

list_checkpoints(CheckpointDir)

load_checkpoint(CheckpointPath)

load_checkpoint(CheckpointPath, Options)

load_distributed_checkpoint(CheckpointDir, NodeId)

load_metadata(FilePath)

load_model(FilePath)

load_model(FilePath, Options)

load_pytorch_state_dict(FilePath)

load_quantized_model(FilePath)

load_safetensors(FilePath)

merge_distributed_checkpoints(CheckpointDir, OutputPath)

save_checkpoint(Model, CheckpointDir)

save_checkpoint(Model, CheckpointDir, Options)

save_distributed_checkpoint(Model, CheckpointDir, NodeId)

save_metadata(Metadata, FilePath)

save_model(Model, FilePath)

save_model(Model, FilePath, Options)

save_pytorch_state_dict(Model, FilePath)

save_quantized_model(Model, QuantParams, FilePath)

save_safetensors(Model, FilePath)

stream_model_weights(StreamState, LayerName)

supported_formats()

verify_safetensors(FilePath)