Index

_

__init__() (mlx.core.array method) (mlx.core.Device method) (mlx.core.Dtype method) (mlx.core.Stream method)

A

abs() (in module mlx.core) (mlx.core.array method) Adam (class in mlx.optimizers) add() (in module mlx.core) all() (in module mlx.core) (mlx.core.array method) allclose() (in module mlx.core) any() (in module mlx.core) (mlx.core.array method) apply() (mlx.nn.Module method) apply_to_modules() (mlx.nn.Module method) arange() (in module mlx.core) arccos() (in module mlx.core)

arccosh() (in module mlx.core) arcsin() (in module mlx.core) arcsinh() (in module mlx.core) arctan() (in module mlx.core) arctanh() (in module mlx.core) argmax() (in module mlx.core) (mlx.core.array method) argmin() (in module mlx.core) (mlx.core.array method) argpartition() (in module mlx.core) argsort() (in module mlx.core) array (class in mlx.core) array_equal() (in module mlx.core) astype() (mlx.core.array method)

B

bernoulli() (in module mlx.core.random)

broadcast_to() (in module mlx.core)

C

categorical() (in module mlx.core.random) children() (mlx.nn.Module method) concatenate() (in module mlx.core) Conv1d (class in mlx.nn) conv1d() (in module mlx.core)

Conv2d (class in mlx.nn) conv2d() (in module mlx.core) convolve() (in module mlx.core) cos() (in module mlx.core) (mlx.core.array method) cosh() (in module mlx.core)

D

default_device() (in module mlx.core) default_stream() (in module mlx.core) Device (class in mlx.core)

divide() (in module mlx.core) Dtype (class in mlx.core) dtype (mlx.core.array property)

E

Embedding (class in mlx.nn) equal() (in module mlx.core) erf() (in module mlx.core) erfinv() (in module mlx.core)

eval() (in module mlx.core) exp() (in module mlx.core) (mlx.core.array method) expand_dims() (in module mlx.core)

F

fft() (in module mlx.core.fft) fft2() (in module mlx.core.fft) fftn() (in module mlx.core.fft)

filter_and_map() (mlx.nn.Module method) freeze() (mlx.nn.Module method) full() (in module mlx.core)

G

GELU (class in mlx.nn) gelu (class in mlx.nn) gelu_approx (class in mlx.nn) gelu_fast_approx (class in mlx.nn)

grad() (in module mlx.core) greater() (in module mlx.core) greater_equal() (in module mlx.core) GroupNorm (class in mlx.nn) gumbel() (in module mlx.core.random)

I

ifft() (in module mlx.core.fft) ifft2() (in module mlx.core.fft) ifftn() (in module mlx.core.fft)

irfft() (in module mlx.core.fft) irfft2() (in module mlx.core.fft) irfftn() (in module mlx.core.fft) item() (mlx.core.array method)

J

jvp() (in module mlx.core)

K

key() (in module mlx.core.random)

L

LayerNorm (class in mlx.nn) leaf_modules() (mlx.nn.Module method) less() (in module mlx.core) less_equal() (in module mlx.core) Linear (class in mlx.nn) load() (in module mlx.core) load_weights() (mlx.nn.Module method) log() (in module mlx.core) (mlx.core.array method)

log10() (in module mlx.core) log1p() (in module mlx.core) (mlx.core.array method) log2() (in module mlx.core) logaddexp() (in module mlx.core) logical_not() (in module mlx.core) logsumexp() (in module mlx.core) (mlx.core.array method)

M

matmul() (in module mlx.core) max() (in module mlx.core) (mlx.core.array method) maximum() (in module mlx.core) mean() (in module mlx.core) (mlx.core.array method)

min() (in module mlx.core) (mlx.core.array method) minimum() (in module mlx.core) Module (class in mlx.nn) modules() (mlx.nn.Module method) MultiHeadAttention (class in mlx.nn) multiply() (in module mlx.core)

N

named_modules() (mlx.nn.Module method) ndim (mlx.core.array property)

negative() (in module mlx.core) new_stream() (in module mlx.core) normal() (in module mlx.core.random)

O

ones() (in module mlx.core) ones_like() (in module mlx.core)

Optimizer (class in mlx.optimizers) OptimizerState (class in mlx.optimizers)

P

pad() (in module mlx.core) parameters() (mlx.nn.Module method)

partition() (in module mlx.core) prod() (in module mlx.core) (mlx.core.array method)

R

randint() (in module mlx.core.random) reciprocal() (in module mlx.core) (mlx.core.array method) ReLU (class in mlx.nn) relu (class in mlx.nn) reshape() (in module mlx.core) (mlx.core.array method)

rfft() (in module mlx.core.fft) rfft2() (in module mlx.core.fft) rfftn() (in module mlx.core.fft) RMSNorm (class in mlx.nn) RoPE (class in mlx.nn) rsqrt() (in module mlx.core) (mlx.core.array method)

S

save() (in module mlx.core) save_weights() (mlx.nn.Module method) savez() (in module mlx.core) savez_compressed() (in module mlx.core) seed() (in module mlx.core.random) Sequential (class in mlx.nn) set_default_device() (in module mlx.core) set_default_stream() (in module mlx.core) SGD (class in mlx.optimizers) shape (mlx.core.array property) sigmoid() (in module mlx.core) sign() (in module mlx.core) SiLU (class in mlx.nn) silu (class in mlx.nn) sin() (in module mlx.core) (mlx.core.array method) sinh() (in module mlx.core)

size (mlx.core.array property) softmax() (in module mlx.core) sort() (in module mlx.core) split() (in module mlx.core) (in module mlx.core.random) (mlx.core.array method) sqrt() (in module mlx.core) (mlx.core.array method) square() (in module mlx.core) (mlx.core.array method) squeeze() (in module mlx.core) state (mlx.optimizers.Optimizer attribute) stop_gradient() (in module mlx.core) Stream (class in mlx.core) subtract() (in module mlx.core) sum() (in module mlx.core) (mlx.core.array method)

T

T (mlx.core.array property) take() (in module mlx.core) take_along_axis() (in module mlx.core) tan() (in module mlx.core) tanh() (in module mlx.core) tolist() (mlx.core.array method)

trainable_parameters() (mlx.nn.Module method) transpose() (in module mlx.core) (mlx.core.array method) tree_flatten() (in module mlx.utils) tree_map() (in module mlx.utils) tree_unflatten() (in module mlx.utils) truncated_normal() (in module mlx.core.random)

U

unfreeze() (mlx.nn.Module method)

uniform() (in module mlx.core.random) update() (mlx.nn.Module method)

V

value_and_grad() (in module mlx.core) (in module mlx.nn) var() (in module mlx.core) (mlx.core.array method)

vjp() (in module mlx.core) vmap() (in module mlx.core)

W

where() (in module mlx.core)

Z

zeros() (in module mlx.core)

zeros_like() (in module mlx.core)