update docs

This commit is contained in:
Awni Hannun
2023-12-13 14:46:24 -08:00
committed by CircleCI Docs
parent f6ae46f713
commit 0f04ebb557
241 changed files with 15049 additions and 737 deletions

View File

@@ -0,0 +1,8 @@
mlx.nn.losses.binary\_cross\_entropy
====================================
.. currentmodule:: mlx.nn.losses
.. autoclass:: binary_cross_entropy

View File

@@ -0,0 +1,8 @@
mlx.nn.losses.cross\_entropy
============================
.. currentmodule:: mlx.nn.losses
.. autoclass:: cross_entropy

View File

@@ -0,0 +1,8 @@
mlx.nn.losses.kl\_div\_loss
===========================
.. currentmodule:: mlx.nn.losses
.. autoclass:: kl_div_loss

View File

@@ -0,0 +1,8 @@
mlx.nn.losses.l1\_loss
======================
.. currentmodule:: mlx.nn.losses
.. autoclass:: l1_loss

View File

@@ -0,0 +1,8 @@
mlx.nn.losses.mse\_loss
=======================
.. currentmodule:: mlx.nn.losses
.. autoclass:: mse_loss

View File

@@ -0,0 +1,8 @@
mlx.nn.losses.nll\_loss
=======================
.. currentmodule:: mlx.nn.losses
.. autoclass:: nll_loss

View File

@@ -0,0 +1,8 @@
mlx.nn.mish
===========
.. currentmodule:: mlx.nn
.. autoclass:: mish

View File

@@ -0,0 +1,8 @@
mlx.nn.prelu
============
.. currentmodule:: mlx.nn
.. autoclass:: prelu

View File

@@ -0,0 +1,8 @@
mlx.nn.selu
===========
.. currentmodule:: mlx.nn
.. autoclass:: selu

View File

@@ -0,0 +1,8 @@
mlx.nn.step
===========
.. currentmodule:: mlx.nn
.. autoclass:: step