mlx
v0.1.0
Search documentation of mlx
Settings
mlx_io
(mlx v0.1.0)
View Source
Summary
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)
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)