MLX Guru - files

2024-01-26 10:03:11 -08:00
parent bd0099ad1f
commit f96f19d404
8 changed files with 1579 additions and 0 deletions
--- a/prompts/gpts/knowledge/MLX
+++ b/prompts/gpts/knowledge/MLX
@@ -0,0 +1,24 @@
+.. _losses:
+
+.. currentmodule:: mlx.nn.losses
+
+Loss Functions
+--------------
+
+.. autosummary::
+   :toctree: _autosummary_functions
+   :template: nn-module-template.rst
+
+   binary_cross_entropy
+   cosine_similarity_loss
+   cross_entropy
+   gaussian_nll_loss
+   hinge_loss
+   huber_loss
+   kl_div_loss
+   l1_loss
+   log_cosh_loss
+   mse_loss
+   nll_loss
+   smooth_l1_loss
+   triplet_loss