docs update

This commit is contained in:
Awni Hannun
2024-05-20 09:40:17 -07:00
committed by CircleCI Docs
parent 93cf6e9f5c
commit 20469ce2d1
528 changed files with 17198 additions and 4162 deletions

View File

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

View File

@@ -0,0 +1,6 @@
mlx.core.as\_strided
====================
.. currentmodule:: mlx.core
.. autofunction:: as_strided

View File

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

View File

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

View File

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

View File

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

View File

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

View File

@@ -8,5 +8,8 @@ Linear Algebra
.. autosummary::
:toctree: _autosummary
inv
norm
cholesky
qr
svd

View File

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

View File

@@ -13,5 +13,6 @@
.. autosummary::
~Embedding.as_linear
~Embedding.to_quantized

View File

@@ -12,5 +12,6 @@
.. autosummary::
~Linear.to_quantized

View File

@@ -15,6 +15,7 @@ Layers
BatchNorm
Conv1d
Conv2d
Conv3d
Dropout
Dropout2d
Dropout3d

View File

@@ -10,6 +10,7 @@ Operations
abs
add
addmm
all
allclose
any
@@ -26,6 +27,7 @@ Operations
argpartition
argsort
array_equal
as_strided
atleast_1d
atleast_2d
atleast_3d
@@ -76,6 +78,7 @@ Operations
isnan
isneginf
isposinf
issubdtype
left_shift
less
less_equal
@@ -106,11 +109,13 @@ Operations
outer
partition
pad
power
prod
quantize
quantized_matmul
radians
reciprocal
remainder
repeat
reshape
right_shift