This commit is contained in:
Awni Hannun
2024-01-10 14:14:38 -08:00
committed by CircleCI Docs
parent 2aeb6df29c
commit 7534da7269
314 changed files with 11870 additions and 691 deletions

View File

@@ -62,6 +62,7 @@
.. autosummary::
~array.T
~array.at
~array.dtype
~array.itemsize
~array.nbytes

View File

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

View File

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

View File

@@ -0,0 +1,6 @@
mlx.core.logical\_and
=====================
.. currentmodule:: mlx.core
.. autofunction:: logical_and

View File

@@ -0,0 +1,6 @@
mlx.core.logical\_or
====================
.. currentmodule:: mlx.core
.. autofunction:: logical_or

View File

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

View File

@@ -0,0 +1,6 @@
mlx.core.save\_gguf
===================
.. currentmodule:: mlx.core
.. autofunction:: save_gguf

View File

@@ -29,9 +29,9 @@ The default floating point type is ``float32`` and the default integer type is
* - ``uint32``
- 4
- 32-bit unsigned integer
* - ``uint32``
* - ``uint64``
- 8
- 32-bit unsigned integer
- 64-bit unsigned integer
* - ``int8``
- 1
- 8-bit signed integer

View File

@@ -0,0 +1,8 @@
mlx.nn.losses.cosine\_similarity\_loss
======================================
.. currentmodule:: mlx.nn.losses
.. autoclass:: cosine_similarity_loss

View File

@@ -19,4 +19,5 @@ Loss Functions
triplet_loss
hinge_loss
huber_loss
log_cosh_loss
log_cosh_loss
cosine_similarity_loss

View File

@@ -36,6 +36,7 @@ Operations
cosh
dequantize
divide
divmod
equal
erf
erfinv
@@ -49,6 +50,7 @@ Operations
greater
greater_equal
identity
inner
less
less_equal
linspace
@@ -59,6 +61,8 @@ Operations
log1p
logaddexp
logical_not
logical_and
logical_or
logsumexp
matmul
max
@@ -71,6 +75,7 @@ Operations
negative
ones
ones_like
outer
partition
pad
prod
@@ -84,6 +89,7 @@ Operations
save
savez
savez_compressed
save_gguf
save_safetensors
sigmoid
sign