mlx_profiler (mlx v0.2.0)

View Source

Summary

Functions

analyze_performance(Model)

auto_tune_parameters(Model, ValidationData)

benchmark_operation(Operation, NumRuns)

benchmark_operation(Operation, NumRuns, Options)

compare_implementations(Implementations, NumRuns)

debug_tensor_flow(Tensors)

detect_memory_leaks()

efficiency_metrics(Model)

find_bottlenecks(ProfileData)

generate_performance_report(AnalysisResults)

get_monitoring_data()

get_profile_stats()

gpu_kernel_stats()

gpu_memory_usage()

gpu_timeline()

gpu_utilization()

latency_analysis(Model, TestInputs)

memory_diff(Snapshot1, Snapshot2)

memory_snapshot()

memory_timeline()

optimize_computation_graph(ComputationGraph)

optimize_computation_graph(ComputationGraph, Options)

profile_computation(Fun)

profile_computation(Fun, Options)

reset_stats()

scalability_analysis(Model, BatchSizes)

start_monitoring()

start_monitoring(Options)

start_profiling()

start_profiling(Options)

stop_monitoring()

stop_profiling()

suggest_optimizations(ProfileData)

throughput_analysis(Model, DataLoader)

trace_execution(Fun)

trace_execution(Fun, Options)

track_memory_usage(Fun)

validate_numerics(Tensor)