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
Using Streams

Examples

Linear Regression
Multi-Layer Perceptron
LLM inference

Python API Reference

Array
Devices and Streams
Operations
Random
Transforms
FFT
Linear Algebra
- mlx.core.linalg.norm
- mlx.core.linalg.qr
Neural Networks
Optimizers
Tree Utils

C++ API Reference

Operations

Further Reading

Developer Documentation

.rst

mlx.core.set_default_device

mlx.core.set_default_device#

mlx.core.set_default_device(device: Device) → None#: Set the default device.

previous

mlx.core.default_device

next

mlx.core.default_stream

By MLX Contributors

© Copyright 2023, MLX Contributors.