docs update

This commit is contained in:
Awni Hannun
2024-04-11 17:33:33 -07:00
committed by CircleCI Docs
parent 8b5550f9f8
commit ba26fe5be5
413 changed files with 9992 additions and 2202 deletions

View File

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

View File

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

View File

@@ -0,0 +1,6 @@
mlx.core.metal.start\_capture
=============================
.. currentmodule:: mlx.core.metal
.. autofunction:: start_capture

View File

@@ -0,0 +1,6 @@
mlx.core.metal.stop\_capture
============================
.. currentmodule:: mlx.core.metal
.. autofunction:: stop_capture

View File

@@ -0,0 +1,6 @@
mlx.core.random.multivariate\_normal
====================================
.. currentmodule:: mlx.core.random
.. autofunction:: multivariate_normal

View File

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

View File

@@ -3,7 +3,7 @@ Metal
.. currentmodule:: mlx.core.metal
.. autosummary::
.. autosummary::
:toctree: _autosummary
is_available
@@ -12,3 +12,5 @@ Metal
get_cache_memory
set_memory_limit
set_cache_limit
start_capture
stop_capture

View File

@@ -5,13 +5,13 @@ Operations
.. currentmodule:: mlx.core
.. autosummary::
.. autosummary::
:toctree: _autosummary
abs
add
all
allclose
allclose
any
arange
arccos
@@ -51,6 +51,7 @@ Operations
erf
erfinv
exp
expm1
expand_dims
eye
flatten
@@ -83,6 +84,7 @@ Operations
max
maximum
mean
meshgrid
min
minimum
moveaxis
@@ -117,6 +119,7 @@ Operations
square
squeeze
stack
std
stop_gradient
subtract
sum

View File

@@ -38,6 +38,7 @@ we use a splittable version of Threefry, which is a counter-based PRNG.
gumbel
key
normal
multivariate_normal
randint
seed
split