Skip to main content

Ctrl+K

Install

Build and Install

Usage

Quick Start Guide
Lazy Evaluation
Unified Memory
Indexing Arrays
Saving and Loading Arrays
Function Transforms
Compilation
Conversion to NumPy and Other Frameworks
Distributed Communication
Using Streams

Examples

Linear Regression
Multi-Layer Perceptron
LLM inference

Python API Reference

Array
Data Types
Devices and Streams
Operations
Random
Transforms
Fast
FFT
Linear Algebra
Metal
Neural Networks
Optimizers
Distributed Communication
Tree Utils

C++ API Reference

Operations

Further Reading

Custom Extensions in MLX
Metal Debugger
Custom Metal Kernels

.rst

mlx.core.metal.reset_peak_memory

Contents

reset_peak_memory()

mlx.core.metal.reset_peak_memory#

reset_peak_memory() → None#: Reset the peak memory to zero.

previous

mlx.core.metal.get_peak_memory

next

mlx.core.metal.get_cache_memory

Contents

reset_peak_memory()

By MLX Contributors

© Copyright 2023, MLX Contributors.