This commit is contained in:
Awni Hannun
2024-09-17 12:06:14 -07:00
committed by CircleCI Docs
parent 9da49a07a4
commit d44f06ae79
739 changed files with 28107 additions and 8524 deletions

View File

@@ -50,6 +50,7 @@
~array.sqrt
~array.square
~array.squeeze
~array.std
~array.sum
~array.swapaxes
~array.tolist

View File

@@ -0,0 +1,6 @@
mlx.core.array.std
==================
.. currentmodule:: mlx.core
.. automethod:: array.std

View File

@@ -0,0 +1,6 @@
mlx.core.conv\_transpose1d
==========================
.. currentmodule:: mlx.core
.. autofunction:: conv_transpose1d

View File

@@ -0,0 +1,6 @@
mlx.core.conv\_transpose2d
==========================
.. currentmodule:: mlx.core
.. autofunction:: conv_transpose2d

View File

@@ -0,0 +1,6 @@
mlx.core.conv\_transpose3d
==========================
.. currentmodule:: mlx.core
.. autofunction:: conv_transpose3d

View File

@@ -0,0 +1,6 @@
mlx.core.distributed.recv
=========================
.. currentmodule:: mlx.core.distributed
.. autofunction:: recv

View File

@@ -0,0 +1,6 @@
mlx.core.distributed.recv\_like
===============================
.. currentmodule:: mlx.core.distributed
.. autofunction:: recv_like

View File

@@ -0,0 +1,6 @@
mlx.core.distributed.send
=========================
.. currentmodule:: mlx.core.distributed
.. autofunction:: send

View File

@@ -3,20 +3,4 @@
.. currentmodule:: mlx.core.fast
.. autoclass:: metal_kernel
.. automethod:: __init__
.. rubric:: Methods
.. autosummary::
~metal_kernel.__init__
.. autofunction:: metal_kernel

View File

@@ -0,0 +1,6 @@
mlx.core.isfinite
=================
.. currentmodule:: mlx.core
.. autofunction:: isfinite

View File

@@ -53,8 +53,9 @@ Array
array.sqrt
array.square
array.squeeze
array.swapaxes
array.std
array.sum
array.swapaxes
array.transpose
array.T
array.var

View File

@@ -17,3 +17,6 @@ made available.
init
all_sum
all_gather
send
recv
recv_like

View File

@@ -0,0 +1,16 @@
mlx.nn.ConvTranspose1d
======================
.. currentmodule:: mlx.nn
.. autoclass:: ConvTranspose1d
.. rubric:: Methods
.. autosummary::

View File

@@ -0,0 +1,16 @@
mlx.nn.ConvTranspose2d
======================
.. currentmodule:: mlx.nn
.. autoclass:: ConvTranspose2d
.. rubric:: Methods
.. autosummary::

View File

@@ -0,0 +1,16 @@
mlx.nn.ConvTranspose3d
======================
.. currentmodule:: mlx.nn
.. autoclass:: ConvTranspose3d
.. rubric:: Methods
.. autosummary::

View File

@@ -16,6 +16,9 @@ Layers
Conv1d
Conv2d
Conv3d
ConvTranspose1d
ConvTranspose2d
ConvTranspose3d
Dropout
Dropout2d
Dropout3d

View File

@@ -45,6 +45,9 @@ Operations
conv1d
conv2d
conv3d
conv_transpose1d
conv_transpose2d
conv_transpose3d
conv_general
cos
cosh
@@ -78,6 +81,7 @@ Operations
hadamard_transform
identity
inner
isfinite
isclose
isinf
isnan