diff --git a/docs/build/html/.buildinfo b/docs/build/html/.buildinfo
index e1d1fbf9..61885dc9 100644
--- a/docs/build/html/.buildinfo
+++ b/docs/build/html/.buildinfo
@@ -1,4 +1,4 @@
# Sphinx build info version 1
# This file hashes the configuration used when building these files. When it is not found, a full rebuild will be done.
-config: 59497ac668d2f3a880c24eaae6f1b8ae
+config: 2a306b1bab538696d27d81f66f046d0d
tags: 645f666f9bcd5a90fca523b33c5a78b7
diff --git a/docs/build/html/_sources/python/_autosummary/mlx.core.atleast_1d.rst b/docs/build/html/_sources/python/_autosummary/mlx.core.atleast_1d.rst
new file mode 100644
index 00000000..4137f119
--- /dev/null
+++ b/docs/build/html/_sources/python/_autosummary/mlx.core.atleast_1d.rst
@@ -0,0 +1,6 @@
+mlx.core.atleast\_1d
+====================
+
+.. currentmodule:: mlx.core
+
+.. autofunction:: atleast_1d
\ No newline at end of file
diff --git a/docs/build/html/_sources/python/_autosummary/mlx.core.atleast_2d.rst b/docs/build/html/_sources/python/_autosummary/mlx.core.atleast_2d.rst
new file mode 100644
index 00000000..94f6c1d9
--- /dev/null
+++ b/docs/build/html/_sources/python/_autosummary/mlx.core.atleast_2d.rst
@@ -0,0 +1,6 @@
+mlx.core.atleast\_2d
+====================
+
+.. currentmodule:: mlx.core
+
+.. autofunction:: atleast_2d
\ No newline at end of file
diff --git a/docs/build/html/_sources/python/_autosummary/mlx.core.atleast_3d.rst b/docs/build/html/_sources/python/_autosummary/mlx.core.atleast_3d.rst
new file mode 100644
index 00000000..ffbc8132
--- /dev/null
+++ b/docs/build/html/_sources/python/_autosummary/mlx.core.atleast_3d.rst
@@ -0,0 +1,6 @@
+mlx.core.atleast\_3d
+====================
+
+.. currentmodule:: mlx.core
+
+.. autofunction:: atleast_3d
\ No newline at end of file
diff --git a/docs/build/html/_sources/python/_autosummary/mlx.core.conv_general.rst b/docs/build/html/_sources/python/_autosummary/mlx.core.conv_general.rst
new file mode 100644
index 00000000..18a35545
--- /dev/null
+++ b/docs/build/html/_sources/python/_autosummary/mlx.core.conv_general.rst
@@ -0,0 +1,6 @@
+mlx.core.conv\_general
+======================
+
+.. currentmodule:: mlx.core
+
+.. autofunction:: conv_general
\ No newline at end of file
diff --git a/docs/build/html/_sources/python/nn/_autosummary/mlx.nn.Upsample.rst b/docs/build/html/_sources/python/nn/_autosummary/mlx.nn.Upsample.rst
new file mode 100644
index 00000000..6bd1855f
--- /dev/null
+++ b/docs/build/html/_sources/python/nn/_autosummary/mlx.nn.Upsample.rst
@@ -0,0 +1,6 @@
+mlx.nn.Upsample
+===============
+
+.. currentmodule:: mlx.nn
+
+.. autoclass:: Upsample
\ No newline at end of file
diff --git a/docs/build/html/_sources/python/nn/_autosummary_functions/mlx.nn.elu.rst b/docs/build/html/_sources/python/nn/_autosummary_functions/mlx.nn.elu.rst
new file mode 100644
index 00000000..7fceaa9d
--- /dev/null
+++ b/docs/build/html/_sources/python/nn/_autosummary_functions/mlx.nn.elu.rst
@@ -0,0 +1,6 @@
+mlx.nn.elu
+==========
+
+.. currentmodule:: mlx.nn
+
+.. autofunction:: elu
\ No newline at end of file
diff --git a/docs/build/html/_sources/python/nn/_autosummary_functions/mlx.nn.glu.rst b/docs/build/html/_sources/python/nn/_autosummary_functions/mlx.nn.glu.rst
new file mode 100644
index 00000000..f289859d
--- /dev/null
+++ b/docs/build/html/_sources/python/nn/_autosummary_functions/mlx.nn.glu.rst
@@ -0,0 +1,6 @@
+mlx.nn.glu
+==========
+
+.. currentmodule:: mlx.nn
+
+.. autofunction:: glu
\ No newline at end of file
diff --git a/docs/build/html/_sources/python/nn/_autosummary_functions/mlx.nn.hardswish.rst b/docs/build/html/_sources/python/nn/_autosummary_functions/mlx.nn.hardswish.rst
new file mode 100644
index 00000000..c824569e
--- /dev/null
+++ b/docs/build/html/_sources/python/nn/_autosummary_functions/mlx.nn.hardswish.rst
@@ -0,0 +1,6 @@
+mlx.nn.hardswish
+================
+
+.. currentmodule:: mlx.nn
+
+.. autofunction:: hardswish
\ No newline at end of file
diff --git a/docs/build/html/_sources/python/nn/_autosummary_functions/mlx.nn.leaky_relu.rst b/docs/build/html/_sources/python/nn/_autosummary_functions/mlx.nn.leaky_relu.rst
new file mode 100644
index 00000000..18f8fdd4
--- /dev/null
+++ b/docs/build/html/_sources/python/nn/_autosummary_functions/mlx.nn.leaky_relu.rst
@@ -0,0 +1,6 @@
+mlx.nn.leaky\_relu
+==================
+
+.. currentmodule:: mlx.nn
+
+.. autofunction:: leaky_relu
\ No newline at end of file
diff --git a/docs/build/html/_sources/python/nn/_autosummary_functions/mlx.nn.log_sigmoid.rst b/docs/build/html/_sources/python/nn/_autosummary_functions/mlx.nn.log_sigmoid.rst
new file mode 100644
index 00000000..662a3a1f
--- /dev/null
+++ b/docs/build/html/_sources/python/nn/_autosummary_functions/mlx.nn.log_sigmoid.rst
@@ -0,0 +1,6 @@
+mlx.nn.log\_sigmoid
+===================
+
+.. currentmodule:: mlx.nn
+
+.. autofunction:: log_sigmoid
\ No newline at end of file
diff --git a/docs/build/html/_sources/python/nn/_autosummary_functions/mlx.nn.log_softmax.rst b/docs/build/html/_sources/python/nn/_autosummary_functions/mlx.nn.log_softmax.rst
new file mode 100644
index 00000000..114bc726
--- /dev/null
+++ b/docs/build/html/_sources/python/nn/_autosummary_functions/mlx.nn.log_softmax.rst
@@ -0,0 +1,6 @@
+mlx.nn.log\_softmax
+===================
+
+.. currentmodule:: mlx.nn
+
+.. autofunction:: log_softmax
\ No newline at end of file
diff --git a/docs/build/html/_sources/python/nn/_autosummary_functions/mlx.nn.relu6.rst b/docs/build/html/_sources/python/nn/_autosummary_functions/mlx.nn.relu6.rst
new file mode 100644
index 00000000..ce7dbb95
--- /dev/null
+++ b/docs/build/html/_sources/python/nn/_autosummary_functions/mlx.nn.relu6.rst
@@ -0,0 +1,6 @@
+mlx.nn.relu6
+============
+
+.. currentmodule:: mlx.nn
+
+.. autofunction:: relu6
\ No newline at end of file
diff --git a/docs/build/html/_sources/python/nn/_autosummary_functions/mlx.nn.sigmoid.rst b/docs/build/html/_sources/python/nn/_autosummary_functions/mlx.nn.sigmoid.rst
new file mode 100644
index 00000000..26e91621
--- /dev/null
+++ b/docs/build/html/_sources/python/nn/_autosummary_functions/mlx.nn.sigmoid.rst
@@ -0,0 +1,6 @@
+mlx.nn.sigmoid
+==============
+
+.. currentmodule:: mlx.nn
+
+.. autofunction:: sigmoid
\ No newline at end of file
diff --git a/docs/build/html/_sources/python/nn/_autosummary_functions/mlx.nn.softmax.rst b/docs/build/html/_sources/python/nn/_autosummary_functions/mlx.nn.softmax.rst
new file mode 100644
index 00000000..6111f807
--- /dev/null
+++ b/docs/build/html/_sources/python/nn/_autosummary_functions/mlx.nn.softmax.rst
@@ -0,0 +1,6 @@
+mlx.nn.softmax
+==============
+
+.. currentmodule:: mlx.nn
+
+.. autofunction:: softmax
\ No newline at end of file
diff --git a/docs/build/html/_sources/python/nn/_autosummary_functions/mlx.nn.softplus.rst b/docs/build/html/_sources/python/nn/_autosummary_functions/mlx.nn.softplus.rst
new file mode 100644
index 00000000..b029eed4
--- /dev/null
+++ b/docs/build/html/_sources/python/nn/_autosummary_functions/mlx.nn.softplus.rst
@@ -0,0 +1,6 @@
+mlx.nn.softplus
+===============
+
+.. currentmodule:: mlx.nn
+
+.. autofunction:: softplus
\ No newline at end of file
diff --git a/docs/build/html/_sources/python/nn/_autosummary_functions/mlx.nn.tanh.rst b/docs/build/html/_sources/python/nn/_autosummary_functions/mlx.nn.tanh.rst
new file mode 100644
index 00000000..45b6f4ac
--- /dev/null
+++ b/docs/build/html/_sources/python/nn/_autosummary_functions/mlx.nn.tanh.rst
@@ -0,0 +1,6 @@
+mlx.nn.tanh
+===========
+
+.. currentmodule:: mlx.nn
+
+.. autofunction:: tanh
\ No newline at end of file
diff --git a/docs/build/html/_sources/python/nn/functions.rst b/docs/build/html/_sources/python/nn/functions.rst
index fc99dcad..db276afd 100644
--- a/docs/build/html/_sources/python/nn/functions.rst
+++ b/docs/build/html/_sources/python/nn/functions.rst
@@ -12,13 +12,24 @@ simple functions.
:toctree: _autosummary_functions
:template: nn-module-template.rst
+ elu
gelu
gelu_approx
gelu_fast_approx
+ glu
+ hardswish
+ leaky_relu
+ log_sigmoid
+ log_softmax
mish
prelu
relu
+ relu6
selu
- softshrink
+ sigmoid
silu
+ softmax
+ softplus
+ softshrink
step
+ tanh
diff --git a/docs/build/html/_sources/python/nn/layers.rst b/docs/build/html/_sources/python/nn/layers.rst
index 0f5fca9d..f6755e8f 100644
--- a/docs/build/html/_sources/python/nn/layers.rst
+++ b/docs/build/html/_sources/python/nn/layers.rst
@@ -40,3 +40,4 @@ Layers
Softshrink
Step
Transformer
+ Upsample
\ No newline at end of file
diff --git a/docs/build/html/_sources/python/ops.rst b/docs/build/html/_sources/python/ops.rst
index 09e2d5f7..2cc2b6d6 100644
--- a/docs/build/html/_sources/python/ops.rst
+++ b/docs/build/html/_sources/python/ops.rst
@@ -25,6 +25,9 @@ Operations
argpartition
argsort
array_equal
+ atleast_1d
+ atleast_2d
+ atleast_3d
broadcast_to
ceil
clip
@@ -32,6 +35,7 @@ Operations
convolve
conv1d
conv2d
+ conv_general
cos
cosh
dequantize
diff --git a/docs/build/html/_sources/python/optimizers/_autosummary/mlx.optimizers.join_schedules.rst b/docs/build/html/_sources/python/optimizers/_autosummary/mlx.optimizers.join_schedules.rst
new file mode 100644
index 00000000..0586e069
--- /dev/null
+++ b/docs/build/html/_sources/python/optimizers/_autosummary/mlx.optimizers.join_schedules.rst
@@ -0,0 +1,6 @@
+mlx.optimizers.join\_schedules
+==============================
+
+.. currentmodule:: mlx.optimizers
+
+.. autofunction:: join_schedules
\ No newline at end of file
diff --git a/docs/build/html/_sources/python/optimizers/_autosummary/mlx.optimizers.linear_schedule.rst b/docs/build/html/_sources/python/optimizers/_autosummary/mlx.optimizers.linear_schedule.rst
new file mode 100644
index 00000000..0a0dee68
--- /dev/null
+++ b/docs/build/html/_sources/python/optimizers/_autosummary/mlx.optimizers.linear_schedule.rst
@@ -0,0 +1,6 @@
+mlx.optimizers.linear\_schedule
+===============================
+
+.. currentmodule:: mlx.optimizers
+
+.. autofunction:: linear_schedule
\ No newline at end of file
diff --git a/docs/build/html/_sources/python/optimizers/schedulers.rst b/docs/build/html/_sources/python/optimizers/schedulers.rst
index a83883dd..50855e1e 100644
--- a/docs/build/html/_sources/python/optimizers/schedulers.rst
+++ b/docs/build/html/_sources/python/optimizers/schedulers.rst
@@ -8,6 +8,8 @@ Schedulers
.. autosummary::
:toctree: _autosummary
- step_decay
- exponential_decay
cosine_decay
+ exponential_decay
+ join_schedules
+ linear_schedule
+ step_decay
diff --git a/docs/build/html/_static/documentation_options.js b/docs/build/html/_static/documentation_options.js
index a117f34f..a168ebdd 100644
--- a/docs/build/html/_static/documentation_options.js
+++ b/docs/build/html/_static/documentation_options.js
@@ -1,6 +1,6 @@
var DOCUMENTATION_OPTIONS = {
URL_ROOT: document.getElementById("documentation_options").getAttribute('data-url_root'),
- VERSION: '0.3.0',
+ VERSION: '0.5.0',
LANGUAGE: 'en',
COLLAPSE_INDEX: false,
BUILDER: 'html',
diff --git a/docs/build/html/_static/mlx_logo.png b/docs/build/html/_static/mlx_logo.png
index 49400bd8..be122bf7 100644
Binary files a/docs/build/html/_static/mlx_logo.png and b/docs/build/html/_static/mlx_logo.png differ
diff --git a/docs/build/html/_static/mlx_logo_dark.png b/docs/build/html/_static/mlx_logo_dark.png
new file mode 100644
index 00000000..cda3c1f6
Binary files /dev/null and b/docs/build/html/_static/mlx_logo_dark.png differ
diff --git a/docs/build/html/cpp/ops.html b/docs/build/html/cpp/ops.html
index 0c9ffa37..0f5b651d 100644
--- a/docs/build/html/cpp/ops.html
+++ b/docs/build/html/cpp/ops.html
@@ -9,7 +9,7 @@
Operations — MLX 0.3.0 documentation
+ Operations — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers Developer Documentation — MLX 0.3.0 documentation
+ Developer Documentation — MLX 0.5.0 documentation
@@ -133,8 +133,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers Linear Regression — MLX 0.3.0 documentation
+ Linear Regression — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers LLM inference — MLX 0.3.0 documentation
+ LLM inference — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers Multi-Layer Perceptron — MLX 0.3.0 documentation
+ Multi-Layer Perceptron — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers Index — MLX 0.3.0 documentation
+ Index — MLX 0.5.0 documentation
@@ -131,8 +131,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers apply_to_modules() (mlx.nn.Module method)
arccos() (in module mlx.core)
arccosh() (in module mlx.core)
@@ -766,6 +784,12 @@ document.write(`
array_equal() (in module mlx.core)
astype() (mlx.core.array method)
+atleast_1d() (in module mlx.core)
+atleast_2d() (in module mlx.core)
+atleast_3d() (in module mlx.core)
AvgPool1d (class in mlx.nn)
Conv2d (class in mlx.nn)
conv2d() (in module mlx.core)
+conv_general() (in module mlx.core)
convolve() (in module mlx.core)
E
glorot_uniform() (in module mlx.nn.init)
+glu() (in module mlx.nn)
grad() (in module mlx.core)
H
+ he_uniform() (in module mlx.nn.init)
+hinge_loss() (in module mlx.nn.losses)
huber_loss() (in module mlx.nn.losses)
@@ -1026,6 +1058,10 @@ document.write(`
J
diff --git a/docs/build/html/index.html b/docs/build/html/index.html
index 002929d6..0a5f4df6 100644
--- a/docs/build/html/index.html
+++ b/docs/build/html/index.html
@@ -9,7 +9,7 @@
MLX — MLX 0.3.0 documentation
+ MLX — MLX 0.5.0 documentation
@@ -133,8 +133,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample
Functions Loss Functions Schedulers Build and Install — MLX 0.3.0 documentation
+ Build and Install — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.Device — MLX 0.3.0 documentation
+ mlx.core.Device — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.Dtype — MLX 0.3.0 documentation
+ mlx.core.Dtype — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.stream — MLX 0.3.0 documentation
+ mlx.core.stream — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.abs — MLX 0.3.0 documentation
+ mlx.core.abs — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.add — MLX 0.3.0 documentation
+ mlx.core.add — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.all — MLX 0.3.0 documentation
+ mlx.core.all — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.allclose — MLX 0.3.0 documentation
+ mlx.core.allclose — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.any — MLX 0.3.0 documentation
+ mlx.core.any — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.arange — MLX 0.3.0 documentation
+ mlx.core.arange — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.arccos — MLX 0.3.0 documentation
+ mlx.core.arccos — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.arccosh — MLX 0.3.0 documentation
+ mlx.core.arccosh — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.arcsin — MLX 0.3.0 documentation
+ mlx.core.arcsin — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.arcsinh — MLX 0.3.0 documentation
+ mlx.core.arcsinh — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.arctan — MLX 0.3.0 documentation
+ mlx.core.arctan — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.arctanh — MLX 0.3.0 documentation
+ mlx.core.arctanh — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.argmax — MLX 0.3.0 documentation
+ mlx.core.argmax — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.argmin — MLX 0.3.0 documentation
+ mlx.core.argmin — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.argpartition — MLX 0.3.0 documentation
+ mlx.core.argpartition — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.argsort — MLX 0.3.0 documentation
+ mlx.core.argsort — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.array.T — MLX 0.3.0 documentation
+ mlx.core.array.T — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.array.abs — MLX 0.3.0 documentation
+ mlx.core.array.abs — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.array.all — MLX 0.3.0 documentation
+ mlx.core.array.all — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.array.any — MLX 0.3.0 documentation
+ mlx.core.array.any — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.array.argmax — MLX 0.3.0 documentation
+ mlx.core.array.argmax — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.array.argmin — MLX 0.3.0 documentation
+ mlx.core.array.argmin — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.array.astype — MLX 0.3.0 documentation
+ mlx.core.array.astype — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.array.cos — MLX 0.3.0 documentation
+ mlx.core.array.cos — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.array.dtype — MLX 0.3.0 documentation
+ mlx.core.array.dtype — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.array.exp — MLX 0.3.0 documentation
+ mlx.core.array.exp — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.array — MLX 0.3.0 documentation
+ mlx.core.array — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.array.item — MLX 0.3.0 documentation
+ mlx.core.array.item — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.array.log — MLX 0.3.0 documentation
+ mlx.core.array.log — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.array.log1p — MLX 0.3.0 documentation
+ mlx.core.array.log1p — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.array.logsumexp — MLX 0.3.0 documentation
+ mlx.core.array.logsumexp — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.array.max — MLX 0.3.0 documentation
+ mlx.core.array.max — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.array.mean — MLX 0.3.0 documentation
+ mlx.core.array.mean — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.array.min — MLX 0.3.0 documentation
+ mlx.core.array.min — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.array.ndim — MLX 0.3.0 documentation
+ mlx.core.array.ndim — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.array.prod — MLX 0.3.0 documentation
+ mlx.core.array.prod — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.array.reciprocal — MLX 0.3.0 documentation
+ mlx.core.array.reciprocal — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.array.reshape — MLX 0.3.0 documentation
+ mlx.core.array.reshape — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.array.round — MLX 0.3.0 documentation
+ mlx.core.array.round — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.array.rsqrt — MLX 0.3.0 documentation
+ mlx.core.array.rsqrt — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.array.shape — MLX 0.3.0 documentation
+ mlx.core.array.shape — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.array.sin — MLX 0.3.0 documentation
+ mlx.core.array.sin — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.array.size — MLX 0.3.0 documentation
+ mlx.core.array.size — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.array.split — MLX 0.3.0 documentation
+ mlx.core.array.split — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.array.sqrt — MLX 0.3.0 documentation
+ mlx.core.array.sqrt — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.array.square — MLX 0.3.0 documentation
+ mlx.core.array.square — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.array.sum — MLX 0.3.0 documentation
+ mlx.core.array.sum — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.array.tolist — MLX 0.3.0 documentation
+ mlx.core.array.tolist — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.array.transpose — MLX 0.3.0 documentation
+ mlx.core.array.transpose — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.array.var — MLX 0.3.0 documentation
+ mlx.core.array.var — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.array_equal — MLX 0.3.0 documentation
+ mlx.core.array_equal — MLX 0.5.0 documentation
@@ -46,7 +46,7 @@
mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers False
next
-
mlx.core.broadcast_to
+
mlx.core.atleast_1d
diff --git a/docs/build/html/python/_autosummary/mlx.core.atleast_1d.html b/docs/build/html/python/_autosummary/mlx.core.atleast_1d.html
new file mode 100644
index 00000000..1590e630
--- /dev/null
+++ b/docs/build/html/python/_autosummary/mlx.core.atleast_1d.html
@@ -0,0 +1,818 @@
+
+
+
+
+
+
+
+
+ mlx.core.atleast_1d — MLX 0.5.0 documentation
+
+
+
+
+
+
+ Skip to main content
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
mlx.core.atleast_1d
+
+
+
+
+
+
+
+
+
+
+mlx.core.atleast_1d
+
+
+mlx.core. atleast_1d ( * arys : array stream : Union [ None , Stream , Device ] = None ) → Union [ array , List [ array ] ]
+Convert all arrays to have at least one dimension.
+
+Parameters:
+
+
+Returns:
+An array or list of arrays with at least one dimension.
+Return type:
+array list array
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
mlx.core.atleast_2d — MLX 0.5.0 documentation
+
+
+
+
+
+
+ Skip to main content
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
mlx.core.atleast_2d
+
+
+
+
+
+
+
+
+
+
+mlx.core.atleast_2d
+
+
+mlx.core. atleast_2d ( * arys : array stream : Union [ None , Stream , Device ] = None ) → Union [ array , List [ array ] ]
+Convert all arrays to have at least two dimensions.
+
+Parameters:
+
+
+Returns:
+An array or list of arrays with at least two dimensions.
+Return type:
+array list array
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
mlx.core.atleast_3d — MLX 0.5.0 documentation
+
+
+
+
+
+
+ Skip to main content
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
mlx.core.atleast_3d
+
+
+
+
+
+
+
+
+
+
+mlx.core.atleast_3d
+
+
+mlx.core. atleast_3d ( * arys : array stream : Union [ None , Stream , Device ] = None ) → Union [ array , List [ array ] ]
+Convert all arrays to have at least three dimensions.
+
+Parameters:
+
+
+Returns:
+An array or list of arrays with at least three dimensions.
+Return type:
+array list array
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
mlx.core.broadcast_to — MLX 0.3.0 documentation
+ mlx.core.broadcast_to — MLX 0.5.0 documentation
@@ -47,7 +47,7 @@
mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.clip — MLX 0.3.0 documentation
+ mlx.core.clip — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.compile — MLX 0.3.0 documentation
+ mlx.core.compile — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers <
list dict array None shapeless (bool , optional ) – A function compiled with the shapeless shapeless shapeless True False
Returns:
diff --git a/docs/build/html/python/_autosummary/mlx.core.concatenate.html b/docs/build/html/python/_autosummary/mlx.core.concatenate.html
index 26550280..364bc870 100644
--- a/docs/build/html/python/_autosummary/mlx.core.concatenate.html
+++ b/docs/build/html/python/_autosummary/mlx.core.concatenate.html
@@ -9,7 +9,7 @@
mlx.core.concatenate — MLX 0.3.0 documentation
+ mlx.core.concatenate — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.conv1d — MLX 0.3.0 documentation
+ mlx.core.conv1d — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.conv2d — MLX 0.3.0 documentation
+ mlx.core.conv2d — MLX 0.5.0 documentation
@@ -46,7 +46,7 @@
mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.conv2d
-mlx.core. conv2d ( input : array weight : array / stride : Union [ int , Tuple [ int , int ] ] = 1 padding : Union [ int , Tuple [ int , int ] ] = 0 dilation : Union [ int , Tuple [ int , int ] ] = 1 groups : Union [ int , Tuple [ int , int ] ] = 1 * stream : Union [ None , Stream , Device ] = None ) → array
+mlx.core. conv2d ( input : array weight : array / stride : Union [ int , Tuple [ int , int ] ] = 1 padding : Union [ int , Tuple [ int , int ] ] = 0 dilation : Union [ int , Tuple [ int , int ] ] = 1 groups : int = 1 * stream : Union [ None , Stream , Device ] = None ) → array 2D convolution over an input with several channels
Note: Only the default groups=1
@@ -744,11 +762,11 @@ if only one number is specified. Default:
next
-
mlx.core.cos
+
mlx.core.conv_general
mlx.core.conv_general — MLX 0.5.0 documentation
+
+
+
+
+
+
+ Skip to main content
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
mlx.core.conv_general
+
+
+
+
+
+
+
+
+
+
+mlx.core.conv_general
+
+
+mlx.core. conv_general ( input : array weight : array / stride : Union [ int , List [ int ] ] = 1 padding : Union [ int , List [ int ] , Tuple [ List [ int ] , List [ int ] ] ] = 0 kernel_dilation : Union [ int , List [ int ] ] = 1 input_dilation : Union [ int , List [ int ] ] = 1 groups : int = 1 flip : bool = false * stream : Union [ None , Stream , Device ] = None ) → array
+General convolution over an input with several channels
+
+
+Parameters:
+
+input (array (N, ..., C_in)
weight (array (C_out, ..., C_in)
stride (int or list ( int ) , optional ) – list 1
padding (int , list ( int ) , or tuple ( list ( int ) , list ( int ) ) , optional ) – list 0
kernel_dilation (int or list ( int ) , optional ) – list 1
input_dilation (int or list ( int ) , optional ) – list 1
groups (int , optional ) – Input feature groups. Default: 1
flip (bool , optional ) – Flip the order in which the spatial dimensions of
+the weights are processed. Performs the cross-correlation operator when
+flip False False
+Returns:
+The convolved array.
+Return type:
+array
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
mlx.core.convolve — MLX 0.3.0 documentation
+ mlx.core.convolve — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.cos — MLX 0.3.0 documentation
+ mlx.core.cos — MLX 0.5.0 documentation
@@ -47,7 +47,7 @@
mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.default_device — MLX 0.3.0 documentation
+ mlx.core.default_device — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.default_stream — MLX 0.3.0 documentation
+ mlx.core.default_stream — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.dequantize — MLX 0.3.0 documentation
+ mlx.core.dequantize — MLX 0.5.0 documentation
@@ -135,8 +135,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.diag — MLX 0.3.0 documentation
+ mlx.core.diag — MLX 0.5.0 documentation
@@ -135,8 +135,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.diagonal — MLX 0.3.0 documentation
+ mlx.core.diagonal — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.disable_compile — MLX 0.3.0 documentation
+ mlx.core.disable_compile — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.divide — MLX 0.3.0 documentation
+ mlx.core.divide — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.divmod — MLX 0.3.0 documentation
+ mlx.core.divmod — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.enable_compile — MLX 0.3.0 documentation
+ mlx.core.enable_compile — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.equal — MLX 0.3.0 documentation
+ mlx.core.equal — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.erf — MLX 0.3.0 documentation
+ mlx.core.erf — MLX 0.5.0 documentation
@@ -135,8 +135,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.erfinv — MLX 0.3.0 documentation
+ mlx.core.erfinv — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.eval — MLX 0.3.0 documentation
+ mlx.core.eval — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.exp — MLX 0.3.0 documentation
+ mlx.core.exp — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.expand_dims — MLX 0.3.0 documentation
+ mlx.core.expand_dims — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.eye — MLX 0.3.0 documentation
+ mlx.core.eye — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.fft.fft — MLX 0.3.0 documentation
+ mlx.core.fft.fft — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.fft.fft2 — MLX 0.3.0 documentation
+ mlx.core.fft.fft2 — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.fft.fftn — MLX 0.3.0 documentation
+ mlx.core.fft.fftn — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.fft.ifft — MLX 0.3.0 documentation
+ mlx.core.fft.ifft — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.fft.ifft2 — MLX 0.3.0 documentation
+ mlx.core.fft.ifft2 — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.fft.ifftn — MLX 0.3.0 documentation
+ mlx.core.fft.ifftn — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.fft.irfft — MLX 0.3.0 documentation
+ mlx.core.fft.irfft — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.fft.irfft2 — MLX 0.3.0 documentation
+ mlx.core.fft.irfft2 — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.fft.irfftn — MLX 0.3.0 documentation
+ mlx.core.fft.irfftn — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.fft.rfft — MLX 0.3.0 documentation
+ mlx.core.fft.rfft — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.fft.rfft2 — MLX 0.3.0 documentation
+ mlx.core.fft.rfft2 — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.fft.rfftn — MLX 0.3.0 documentation
+ mlx.core.fft.rfftn — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.flatten — MLX 0.3.0 documentation
+ mlx.core.flatten — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.floor — MLX 0.3.0 documentation
+ mlx.core.floor — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.floor_divide — MLX 0.3.0 documentation
+ mlx.core.floor_divide — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.full — MLX 0.3.0 documentation
+ mlx.core.full — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.grad — MLX 0.3.0 documentation
+ mlx.core.grad — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.greater — MLX 0.3.0 documentation
+ mlx.core.greater — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.greater_equal — MLX 0.3.0 documentation
+ mlx.core.greater_equal — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.identity — MLX 0.3.0 documentation
+ mlx.core.identity — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.inner — MLX 0.3.0 documentation
+ mlx.core.inner — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.isinf — MLX 0.3.0 documentation
+ mlx.core.isinf — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.isnan — MLX 0.3.0 documentation
+ mlx.core.isnan — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.isneginf — MLX 0.3.0 documentation
+ mlx.core.isneginf — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.isposinf — MLX 0.3.0 documentation
+ mlx.core.isposinf — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.jvp — MLX 0.3.0 documentation
+ mlx.core.jvp — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.less — MLX 0.3.0 documentation
+ mlx.core.less — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.less_equal — MLX 0.3.0 documentation
+ mlx.core.less_equal — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.linalg.norm — MLX 0.3.0 documentation
+ mlx.core.linalg.norm — MLX 0.5.0 documentation
@@ -135,8 +135,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.linalg.qr — MLX 0.3.0 documentation
+ mlx.core.linalg.qr — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.linspace — MLX 0.3.0 documentation
+ mlx.core.linspace — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.load — MLX 0.3.0 documentation
+ mlx.core.load — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.log — MLX 0.3.0 documentation
+ mlx.core.log — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.log10 — MLX 0.3.0 documentation
+ mlx.core.log10 — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.log1p — MLX 0.3.0 documentation
+ mlx.core.log1p — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.log2 — MLX 0.3.0 documentation
+ mlx.core.log2 — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.logaddexp — MLX 0.3.0 documentation
+ mlx.core.logaddexp — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.logical_and — MLX 0.3.0 documentation
+ mlx.core.logical_and — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.logical_not — MLX 0.3.0 documentation
+ mlx.core.logical_not — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.logical_or — MLX 0.3.0 documentation
+ mlx.core.logical_or — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.logsumexp — MLX 0.3.0 documentation
+ mlx.core.logsumexp — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.matmul — MLX 0.3.0 documentation
+ mlx.core.matmul — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.max — MLX 0.3.0 documentation
+ mlx.core.max — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.maximum — MLX 0.3.0 documentation
+ mlx.core.maximum — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.mean — MLX 0.3.0 documentation
+ mlx.core.mean — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.min — MLX 0.3.0 documentation
+ mlx.core.min — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.minimum — MLX 0.3.0 documentation
+ mlx.core.minimum — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.moveaxis — MLX 0.3.0 documentation
+ mlx.core.moveaxis — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.multiply — MLX 0.3.0 documentation
+ mlx.core.multiply — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.negative — MLX 0.3.0 documentation
+ mlx.core.negative — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.new_stream — MLX 0.3.0 documentation
+ mlx.core.new_stream — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.ones — MLX 0.3.0 documentation
+ mlx.core.ones — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.ones_like — MLX 0.3.0 documentation
+ mlx.core.ones_like — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.outer — MLX 0.3.0 documentation
+ mlx.core.outer — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.pad — MLX 0.3.0 documentation
+ mlx.core.pad — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.partition — MLX 0.3.0 documentation
+ mlx.core.partition — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.prod — MLX 0.3.0 documentation
+ mlx.core.prod — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.quantize — MLX 0.3.0 documentation
+ mlx.core.quantize — MLX 0.5.0 documentation
@@ -135,8 +135,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.quantized_matmul — MLX 0.3.0 documentation
+ mlx.core.quantized_matmul — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.random.bernoulli — MLX 0.3.0 documentation
+ mlx.core.random.bernoulli — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.random.categorical — MLX 0.3.0 documentation
+ mlx.core.random.categorical — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.random.gumbel — MLX 0.3.0 documentation
+ mlx.core.random.gumbel — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.random.key — MLX 0.3.0 documentation
+ mlx.core.random.key — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.random.normal — MLX 0.3.0 documentation
+ mlx.core.random.normal — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.random.randint — MLX 0.3.0 documentation
+ mlx.core.random.randint — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.random.seed — MLX 0.3.0 documentation
+ mlx.core.random.seed — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.random.split — MLX 0.3.0 documentation
+ mlx.core.random.split — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.random.truncated_normal — MLX 0.3.0 documentation
+ mlx.core.random.truncated_normal — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.random.uniform — MLX 0.3.0 documentation
+ mlx.core.random.uniform — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.reciprocal — MLX 0.3.0 documentation
+ mlx.core.reciprocal — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.repeat — MLX 0.3.0 documentation
+ mlx.core.repeat — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.reshape — MLX 0.3.0 documentation
+ mlx.core.reshape — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.round — MLX 0.3.0 documentation
+ mlx.core.round — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.rsqrt — MLX 0.3.0 documentation
+ mlx.core.rsqrt — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.save — MLX 0.3.0 documentation
+ mlx.core.save — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.save_gguf — MLX 0.3.0 documentation
+ mlx.core.save_gguf — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.save_safetensors — MLX 0.3.0 documentation
+ mlx.core.save_safetensors — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.savez — MLX 0.3.0 documentation
+ mlx.core.savez — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.savez_compressed — MLX 0.3.0 documentation
+ mlx.core.savez_compressed — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.set_default_device — MLX 0.3.0 documentation
+ mlx.core.set_default_device — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.set_default_stream — MLX 0.3.0 documentation
+ mlx.core.set_default_stream — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.sigmoid — MLX 0.3.0 documentation
+ mlx.core.sigmoid — MLX 0.5.0 documentation
@@ -135,8 +135,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.sign — MLX 0.3.0 documentation
+ mlx.core.sign — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.sin — MLX 0.3.0 documentation
+ mlx.core.sin — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.sinh — MLX 0.3.0 documentation
+ mlx.core.sinh — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.softmax — MLX 0.3.0 documentation
+ mlx.core.softmax — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.sort — MLX 0.3.0 documentation
+ mlx.core.sort — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.split — MLX 0.3.0 documentation
+ mlx.core.split — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.sqrt — MLX 0.3.0 documentation
+ mlx.core.sqrt — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.square — MLX 0.3.0 documentation
+ mlx.core.square — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.squeeze — MLX 0.3.0 documentation
+ mlx.core.squeeze — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.stack — MLX 0.3.0 documentation
+ mlx.core.stack — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.stop_gradient — MLX 0.3.0 documentation
+ mlx.core.stop_gradient — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.subtract — MLX 0.3.0 documentation
+ mlx.core.subtract — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.sum — MLX 0.3.0 documentation
+ mlx.core.sum — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.swapaxes — MLX 0.3.0 documentation
+ mlx.core.swapaxes — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.take — MLX 0.3.0 documentation
+ mlx.core.take — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.take_along_axis — MLX 0.3.0 documentation
+ mlx.core.take_along_axis — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.tan — MLX 0.3.0 documentation
+ mlx.core.tan — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.tanh — MLX 0.3.0 documentation
+ mlx.core.tanh — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.tensordot — MLX 0.3.0 documentation
+ mlx.core.tensordot — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.transpose — MLX 0.3.0 documentation
+ mlx.core.transpose — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.tri — MLX 0.3.0 documentation
+ mlx.core.tri — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.tril — MLX 0.3.0 documentation
+ mlx.core.tril — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.triu — MLX 0.3.0 documentation
+ mlx.core.triu — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.value_and_grad — MLX 0.3.0 documentation
+ mlx.core.value_and_grad — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.var — MLX 0.3.0 documentation
+ mlx.core.var — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.vjp — MLX 0.3.0 documentation
+ mlx.core.vjp — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.vmap — MLX 0.3.0 documentation
+ mlx.core.vmap — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.where — MLX 0.3.0 documentation
+ mlx.core.where — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.zeros — MLX 0.3.0 documentation
+ mlx.core.zeros — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.zeros_like — MLX 0.3.0 documentation
+ mlx.core.zeros_like — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.nn.value_and_grad — MLX 0.3.0 documentation
+ mlx.nn.value_and_grad — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.utils.tree_flatten — MLX 0.3.0 documentation
+ mlx.utils.tree_flatten — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.utils.tree_map — MLX 0.3.0 documentation
+ mlx.utils.tree_map — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.utils.tree_unflatten — MLX 0.3.0 documentation
+ mlx.utils.tree_unflatten — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.core.Stream — MLX 0.3.0 documentation
+ mlx.core.Stream — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers Array — MLX 0.3.0 documentation
+ Array — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers Data Types — MLX 0.3.0 documentation
+ Data Types — MLX 0.5.0 documentation
@@ -132,8 +132,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers Devices and Streams — MLX 0.3.0 documentation
+ Devices and Streams — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers FFT — MLX 0.3.0 documentation
+ FFT — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers Linear Algebra — MLX 0.3.0 documentation
+ Linear Algebra — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers Neural Networks — MLX 0.3.0 documentation
+ Neural Networks — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions
Loss Functions
diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.ALiBi.html b/docs/build/html/python/nn/_autosummary/mlx.nn.ALiBi.html
index 9434770b..ae82f1b6 100644
--- a/docs/build/html/python/nn/_autosummary/mlx.nn.ALiBi.html
+++ b/docs/build/html/python/nn/_autosummary/mlx.nn.ALiBi.html
@@ -9,7 +9,7 @@
mlx.nn.ALiBi — MLX 0.3.0 documentation
+ mlx.nn.ALiBi — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample
Functions Loss Functions Schedulers mlx.nn.AvgPool1d — MLX 0.3.0 documentation
+ mlx.nn.AvgPool1d — MLX 0.5.0 documentation
@@ -135,8 +135,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.nn.AvgPool2d — MLX 0.3.0 documentation
+ mlx.nn.AvgPool2d — MLX 0.5.0 documentation
@@ -135,8 +135,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.nn.BatchNorm — MLX 0.3.0 documentation
+ mlx.nn.BatchNorm — MLX 0.5.0 documentation
@@ -135,8 +135,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.nn.Conv1d — MLX 0.3.0 documentation
+ mlx.nn.Conv1d — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.nn.Conv2d — MLX 0.3.0 documentation
+ mlx.nn.Conv2d — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.nn.Dropout — MLX 0.3.0 documentation
+ mlx.nn.Dropout — MLX 0.5.0 documentation
@@ -135,8 +135,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.nn.Dropout2d — MLX 0.3.0 documentation
+ mlx.nn.Dropout2d — MLX 0.5.0 documentation
@@ -135,8 +135,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.nn.Dropout3d — MLX 0.3.0 documentation
+ mlx.nn.Dropout3d — MLX 0.5.0 documentation
@@ -135,8 +135,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.nn.Embedding — MLX 0.3.0 documentation
+ mlx.nn.Embedding — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.nn.GELU — MLX 0.3.0 documentation
+ mlx.nn.GELU — MLX 0.5.0 documentation
@@ -135,8 +135,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.nn.GroupNorm — MLX 0.3.0 documentation
+ mlx.nn.GroupNorm — MLX 0.5.0 documentation
@@ -135,8 +135,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.nn.InstanceNorm — MLX 0.3.0 documentation
+ mlx.nn.InstanceNorm — MLX 0.5.0 documentation
@@ -135,8 +135,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.nn.LayerNorm — MLX 0.3.0 documentation
+ mlx.nn.LayerNorm — MLX 0.5.0 documentation
@@ -135,8 +135,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.nn.Linear — MLX 0.3.0 documentation
+ mlx.nn.Linear — MLX 0.5.0 documentation
@@ -135,8 +135,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.nn.MaxPool1d — MLX 0.3.0 documentation
+ mlx.nn.MaxPool1d — MLX 0.5.0 documentation
@@ -135,8 +135,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.nn.MaxPool2d — MLX 0.3.0 documentation
+ mlx.nn.MaxPool2d — MLX 0.5.0 documentation
@@ -135,8 +135,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.nn.Mish — MLX 0.3.0 documentation
+ mlx.nn.Mish — MLX 0.5.0 documentation
@@ -135,8 +135,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers
class mlx.nn. Mish
-Applies the Mish function, element-wise.
-Mish: A Self Regularized Non-Monotonic Neural Activation Function.
+Applies the Mish function, element-wise.
Reference: https://arxiv.org/abs/1908.08681
\[\text{Mish}(x) = x * \text{Tanh}(\text{Softplus}(x))\]
diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.apply.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.apply.html
index 83226097..f521fab8 100644
--- a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.apply.html
+++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.apply.html
@@ -9,7 +9,7 @@
mlx.nn.Module.apply — MLX 0.3.0 documentation
+ mlx.nn.Module.apply — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.nn.Module.apply_to_modules — MLX 0.3.0 documentation
+ mlx.nn.Module.apply_to_modules — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.nn.Module.children — MLX 0.3.0 documentation
+ mlx.nn.Module.children — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.nn.Module.eval — MLX 0.3.0 documentation
+ mlx.nn.Module.eval — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.nn.Module.filter_and_map — MLX 0.3.0 documentation
+ mlx.nn.Module.filter_and_map — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.nn.Module.freeze — MLX 0.3.0 documentation
+ mlx.nn.Module.freeze — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.nn.Module.leaf_modules — MLX 0.3.0 documentation
+ mlx.nn.Module.leaf_modules — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.nn.Module.load_weights — MLX 0.3.0 documentation
+ mlx.nn.Module.load_weights — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.nn.Module.modules — MLX 0.3.0 documentation
+ mlx.nn.Module.modules — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.nn.Module.named_modules — MLX 0.3.0 documentation
+ mlx.nn.Module.named_modules — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.nn.Module.parameters — MLX 0.3.0 documentation
+ mlx.nn.Module.parameters — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.nn.Module.save_weights — MLX 0.3.0 documentation
+ mlx.nn.Module.save_weights — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.nn.Module.state — MLX 0.3.0 documentation
+ mlx.nn.Module.state — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.nn.Module.train — MLX 0.3.0 documentation
+ mlx.nn.Module.train — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.nn.Module.trainable_parameters — MLX 0.3.0 documentation
+ mlx.nn.Module.trainable_parameters — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.nn.Module.training — MLX 0.3.0 documentation
+ mlx.nn.Module.training — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.nn.Module.unfreeze — MLX 0.3.0 documentation
+ mlx.nn.Module.unfreeze — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.nn.Module.update — MLX 0.3.0 documentation
+ mlx.nn.Module.update — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.nn.Module.update_modules — MLX 0.3.0 documentation
+ mlx.nn.Module.update_modules — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.nn.MultiHeadAttention — MLX 0.3.0 documentation
+ mlx.nn.MultiHeadAttention — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.nn.PReLU — MLX 0.3.0 documentation
+ mlx.nn.PReLU — MLX 0.5.0 documentation
@@ -135,8 +135,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.nn.QuantizedLinear — MLX 0.3.0 documentation
+ mlx.nn.QuantizedLinear — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.nn.RMSNorm — MLX 0.3.0 documentation
+ mlx.nn.RMSNorm — MLX 0.5.0 documentation
@@ -135,8 +135,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.nn.ReLU — MLX 0.3.0 documentation
+ mlx.nn.ReLU — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers
class mlx.nn. ReLU
-Applies the Rectified Linear Unit.
-Simply mx.maximum(x, 0)
+
+Applies the Rectified Linear Unit. Simply mx.maximum(x, 0)
+
+See relu()
diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.RoPE.html b/docs/build/html/python/nn/_autosummary/mlx.nn.RoPE.html
index 07b7cb69..801ba901 100644
--- a/docs/build/html/python/nn/_autosummary/mlx.nn.RoPE.html
+++ b/docs/build/html/python/nn/_autosummary/mlx.nn.RoPE.html
@@ -9,7 +9,7 @@
mlx.nn.RoPE — MLX 0.3.0 documentation
+ mlx.nn.RoPE — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.nn.SELU — MLX 0.3.0 documentation
+ mlx.nn.SELU — MLX 0.5.0 documentation
@@ -43,7 +43,6 @@
-
mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers
class mlx.nn. SELU
Applies the Scaled Exponential Linear Unit.
-
-\[\begin{split}\text{selu}(x) = \begin{cases}
-\lambda x & \text{if } x > 0 \\
-\lambda \alpha (\exp(x) - 1) & \text{if } x \leq 0
-\end{cases}\end{split}\]
-where \(\lambda = 1.0507\) and \(\alpha = 1.67326\) .
-See also elu()
+See selu()
diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Sequential.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Sequential.html
index 5b2f5bf6..a135b234 100644
--- a/docs/build/html/python/nn/_autosummary/mlx.nn.Sequential.html
+++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Sequential.html
@@ -9,7 +9,7 @@
mlx.nn.Sequential — MLX 0.3.0 documentation
+ mlx.nn.Sequential — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.nn.SiLU — MLX 0.3.0 documentation
+ mlx.nn.SiLU — MLX 0.5.0 documentation
@@ -43,7 +43,6 @@
-
mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers
class mlx.nn. SiLU
Applies the Sigmoid Linear Unit. Also known as Swish.
-Applies \(x \sigma(x)\) element wise, where \(\sigma(\cdot)\) is
-the logistic sigmoid.
+See silu()
diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.SinusoidalPositionalEncoding.html b/docs/build/html/python/nn/_autosummary/mlx.nn.SinusoidalPositionalEncoding.html
index 99d2abc2..2863cf76 100644
--- a/docs/build/html/python/nn/_autosummary/mlx.nn.SinusoidalPositionalEncoding.html
+++ b/docs/build/html/python/nn/_autosummary/mlx.nn.SinusoidalPositionalEncoding.html
@@ -9,7 +9,7 @@
mlx.nn.SinusoidalPositionalEncoding — MLX 0.3.0 documentation
+ mlx.nn.SinusoidalPositionalEncoding — MLX 0.5.0 documentation
@@ -135,8 +135,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.nn.Softshrink — MLX 0.3.0 documentation
+ mlx.nn.Softshrink — MLX 0.5.0 documentation
@@ -135,8 +135,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.nn.Step — MLX 0.3.0 documentation
+ mlx.nn.Step — MLX 0.5.0 documentation
@@ -135,8 +135,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.nn.Transformer — MLX 0.3.0 documentation
+ mlx.nn.Transformer — MLX 0.5.0 documentation
@@ -46,7 +46,7 @@
mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.nn.Transformer
-class mlx.nn. Transformer ( dims: int = 512, num_heads: int = 8, num_encoder_layers: int = 6, num_decoder_layers: int = 6, mlp_dims: ~typing.Optional[int] = None, dropout: float = 0.0, activation: ~typing.Callable[[~typing.Any], ~typing.Any] = <function relu>, custom_encoder: ~typing.Optional[~typing.Any] = None, custom_decoder: ~typing.Optional[~typing.Any] = None, norm_first: bool = True, checkpoint: bool = False )
+class mlx.nn. Transformer ( dims: int = 512, num_heads: int = 8, num_encoder_layers: int = 6, num_decoder_layers: int = 6, mlp_dims: ~typing.Optional[int] = None, dropout: float = 0.0, activation: ~typing.Callable[[~typing.Any], ~typing.Any] = <built-in method relu of PyCapsule object>, custom_encoder: ~typing.Optional[~typing.Any] = None, custom_decoder: ~typing.Optional[~typing.Any] = None, norm_first: bool = True, checkpoint: bool = False )
Implements a standard Transformer model.
The implementation is based on Attention Is All You Need .
The Transformer model contains an encoder and a decoder. The encoder
@@ -756,11 +774,11 @@ Default: False
next
-
Functions
+
mlx.nn.Upsample
diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Upsample.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Upsample.html
new file mode 100644
index 00000000..b07fd1f7
--- /dev/null
+++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Upsample.html
@@ -0,0 +1,864 @@
+
+
+
+
+
+
+
+
+ mlx.nn.Upsample — MLX 0.5.0 documentation
+
+
+
+
+
+
+ Skip to main content
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+mlx.nn.Upsample
+
+
+class mlx.nn. Upsample ( scale_factor : Union [ float , Tuple ] mode : Literal [ 'nearest' , 'linear' ] = 'nearest' align_corners : bool = False )
+Upsample the input signal spatially.
+The spatial dimensions are by convention dimensions 1 x.ndim -
+2
+For example, an audio signal would be 3D with 1 spatial dimension, an image
+4D with 2 and so on and so forth.
+There are two upsampling algorithms implemented nearest neighbor upsampling
+and linear interpolation. Both can be applied to any number of spatial
+dimensions and the linear interpolation will be bilinear, trilinear etc
+when applied to more than one spatial dimension.
+
+
Note
+
When using one of the linear interpolation modes the align_corners align_corners=True
+
+Parameters:
+
+scale_factor (float or tuple float
mode (str , optional ) – The upsampling algorithm, either "nearest" "linear" "nearest"
align_corners (bool , optional ) – Changes the way the corners are treated
+during "linear" False
+
+Examples
+>>> import mlx.core as mx
+>>> import mlx.nn as nn
+>>> x = mx . arange ( 1 , 5 ) . reshape (( 1 , 2 , 2 , 1 ))
+>>> x
+array([[[[1],
+ [2]],
+ [[3],
+ [4]]]], dtype=int32)
+>>> n = nn . Upsample ( scale_factor = 2 , mode = 'nearest' )
+>>> n ( x ) . squeeze ()
+array([[1, 1, 2, 2],
+ [1, 1, 2, 2],
+ [3, 3, 4, 4],
+ [3, 3, 4, 4]], dtype=int32)
+>>> b = nn . Upsample ( scale_factor = 2 , mode = 'linear' )
+>>> b ( x ) . squeeze ()
+array([[1, 1.25, 1.75, 2],
+ [1.5, 1.75, 2.25, 2.5],
+ [2.5, 2.75, 3.25, 3.5],
+ [3, 3.25, 3.75, 4]], dtype=float32)
+>>> b = nn . Upsample ( scale_factor = 2 , mode = 'linear' , align_corners = True )
+>>> b ( x ) . squeeze ()
+array([[1, 1.33333, 1.66667, 2],
+ [1.66667, 2, 2.33333, 2.66667],
+ [2.33333, 2.66667, 3, 3.33333],
+ [3, 3.33333, 3.66667, 4]], dtype=float32)
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
mlx.nn.init.constant — MLX 0.3.0 documentation
+ mlx.nn.init.constant — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.nn.init.glorot_normal — MLX 0.3.0 documentation
+ mlx.nn.init.glorot_normal — MLX 0.5.0 documentation
@@ -135,8 +135,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.nn.init.glorot_uniform — MLX 0.3.0 documentation
+ mlx.nn.init.glorot_uniform — MLX 0.5.0 documentation
@@ -135,8 +135,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.nn.init.he_normal — MLX 0.3.0 documentation
+ mlx.nn.init.he_normal — MLX 0.5.0 documentation
@@ -135,8 +135,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.nn.init.he_uniform — MLX 0.3.0 documentation
+ mlx.nn.init.he_uniform — MLX 0.5.0 documentation
@@ -135,8 +135,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.nn.init.identity — MLX 0.3.0 documentation
+ mlx.nn.init.identity — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.nn.init.normal — MLX 0.3.0 documentation
+ mlx.nn.init.normal — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.nn.init.uniform — MLX 0.3.0 documentation
+ mlx.nn.init.uniform — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.nn.elu — MLX 0.5.0 documentation
+
+
+
+
+
+
+ Skip to main content
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+mlx.nn.elu
+
+
+mlx.nn. elu ( x alpha = 1.0 )
+Applies the Exponential Linear Unit.
+Simply mx.where(x > 0, x, alpha * (mx.exp(x) - 1))
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
mlx.nn.gelu — MLX 0.3.0 documentation
+ mlx.nn.gelu — MLX 0.5.0 documentation
@@ -48,7 +48,7 @@
mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.nn.gelu
-mlx.nn. gelu ( x )
+mlx.nn. gelu ( x ) → array Applies the Gaussian Error Linear Units function.
-\[\begin{split}\\textrm{GELU}(x) = x * \Phi(x)\end{split}\]
+\[\textrm{GELU}(x) = x * \Phi(x)\]
where \(\Phi(x)\) is the Gaussian CDF.
See also gelu_approx() gelu_fast_approx()
@@ -716,12 +734,12 @@ approximations.
@@ -234,6 +234,9 @@
mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.nn.gelu_fast_approx — MLX 0.3.0 documentation
+ mlx.nn.gelu_fast_approx — MLX 0.5.0 documentation
@@ -47,7 +47,7 @@
mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers This function approximates gelu \(<
0.015\) in the range \([-6, 6]\) using the following
-\[x = x \sigma\left(1.773 x\right)\]
+\[x = x \sigma\left(1.702 x\right)\]
where \(\sigma(\cdot)\) is the logistic sigmoid.
+References:
+- hendrycks/GELUs
+- https://arxiv.org/abs/1606.08415
@@ -726,11 +747,11 @@ document.write(`
next
-
mlx.nn.mish
+
mlx.nn.glu
diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.glu.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.glu.html
new file mode 100644
index 00000000..5a46e27b
--- /dev/null
+++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.glu.html
@@ -0,0 +1,814 @@
+
+
+
+
+
+
+
+
+ mlx.nn.glu — MLX 0.5.0 documentation
+
+
+
+
+
+
+ Skip to main content
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+mlx.nn.glu
+
+
+mlx.nn. glu ( x : array axis : int = - 1 ) → array
+Applies the gated linear unit function.
+This function splits the axis \(a\) and \(b\) ) and applies \(a * \sigma(b)\) .
+
+\[\textrm{GLU}(x) = a * \sigma(b)\]
+
+Parameters:
+axis (int -1
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
mlx.nn.hardswish — MLX 0.5.0 documentation
+
+
+
+
+
+
+ Skip to main content
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+mlx.nn.hardswish
+
+
+mlx.nn. hardswish ( x )
+Applies the hardswish function, element-wise.
+
+\[\text{Hardswish}(x) = x * \min(\max(x + 3, 0), 6) / 6\]
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
mlx.nn.leaky_relu — MLX 0.5.0 documentation
+
+
+
+
+
+
+ Skip to main content
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
mlx.nn.leaky_relu
+
+
+
+
+
+
+
+
+
+
+mlx.nn.leaky_relu
+
+
+mlx.nn. leaky_relu ( x negative_slope = 0.01 )
+Applies the Leaky Rectified Linear Unit.
+Simply mx.maximum(negative_slope * x, x)
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
mlx.nn.log_sigmoid — MLX 0.5.0 documentation
+
+
+
+
+
+
+ Skip to main content
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
mlx.nn.log_sigmoid
+
+
+
+
+
+
+
+
+
+
+mlx.nn.log_sigmoid
+
+
+mlx.nn. log_sigmoid ( x )
+Applies the Log Sigmoid function.
+Applies \(\log(\sigma(x)) = -\log(1 + e^{-x})\) element wise.
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
mlx.nn.log_softmax — MLX 0.5.0 documentation
+
+
+
+
+
+
+ Skip to main content
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
mlx.nn.log_softmax
+
+
+
+
+
+
+
+
+
+
+mlx.nn.log_softmax
+
+
+mlx.nn. log_softmax ( x axis = - 1 )
+Applies the Log Softmax function.
+Applies \(x + \log \sum_i e^{x_i}\) element wise.
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
mlx.nn.losses.binary_cross_entropy — MLX 0.3.0 documentation
+ mlx.nn.losses.binary_cross_entropy — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.nn.losses.cosine_similarity_loss — MLX 0.3.0 documentation
+ mlx.nn.losses.cosine_similarity_loss — MLX 0.5.0 documentation
@@ -135,8 +135,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.nn.losses.cross_entropy — MLX 0.3.0 documentation
+ mlx.nn.losses.cross_entropy — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.nn.losses.gaussian_nll_loss — MLX 0.3.0 documentation
+ mlx.nn.losses.gaussian_nll_loss — MLX 0.5.0 documentation
@@ -135,8 +135,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.nn.losses.hinge_loss — MLX 0.3.0 documentation
+ mlx.nn.losses.hinge_loss — MLX 0.5.0 documentation
@@ -135,8 +135,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.nn.losses.huber_loss — MLX 0.3.0 documentation
+ mlx.nn.losses.huber_loss — MLX 0.5.0 documentation
@@ -135,8 +135,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.nn.losses.kl_div_loss — MLX 0.3.0 documentation
+ mlx.nn.losses.kl_div_loss — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.nn.losses.l1_loss — MLX 0.3.0 documentation
+ mlx.nn.losses.l1_loss — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.nn.losses.log_cosh_loss — MLX 0.3.0 documentation
+ mlx.nn.losses.log_cosh_loss — MLX 0.5.0 documentation
@@ -135,8 +135,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.nn.losses.margin_ranking_loss — MLX 0.3.0 documentation
+ mlx.nn.losses.margin_ranking_loss — MLX 0.5.0 documentation
@@ -135,8 +135,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.nn.losses.mse_loss — MLX 0.3.0 documentation
+ mlx.nn.losses.mse_loss — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.nn.losses.nll_loss — MLX 0.3.0 documentation
+ mlx.nn.losses.nll_loss — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.nn.losses.smooth_l1_loss — MLX 0.3.0 documentation
+ mlx.nn.losses.smooth_l1_loss — MLX 0.5.0 documentation
@@ -135,8 +135,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.nn.losses.triplet_loss — MLX 0.3.0 documentation
+ mlx.nn.losses.triplet_loss — MLX 0.5.0 documentation
@@ -135,8 +135,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.nn.mish — MLX 0.3.0 documentation
+ mlx.nn.mish — MLX 0.5.0 documentation
@@ -48,7 +48,7 @@
mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers
mlx.nn. mish ( x : array ) → array
-Applies the Mish function, element-wise.
-Mish: A Self Regularized Non-Monotonic Neural Activation Function.
+Applies the Mish function, element-wise.
+Mish: A Self Regularized Non-Monotonic Neural Activation Function.
Reference: https://arxiv.org/abs/1908.08681
\[\text{Mish}(x) = x * \text{Tanh}(\text{Softplus}(x))\]
@@ -715,12 +733,12 @@ Mish: A Self Regularized Non-Monotonic Neural Activation Function.
@@ -234,6 +234,9 @@
mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.nn.relu — MLX 0.3.0 documentation
+ mlx.nn.relu — MLX 0.5.0 documentation
@@ -46,7 +46,7 @@
mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers
next
-
mlx.nn.selu
+
mlx.nn.relu6
diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.relu6.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.relu6.html
new file mode 100644
index 00000000..8569173c
--- /dev/null
+++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.relu6.html
@@ -0,0 +1,806 @@
+
+
+
+
+
+
+
+
+ mlx.nn.relu6 — MLX 0.5.0 documentation
+
+
+
+
+
+
+ Skip to main content
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+mlx.nn.relu6
+
+
+mlx.nn. relu6 ( x )
+Applies the Rectified Linear Unit 6.
+Applies \(\min(\max(x, 0), 6)\) element wise.
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
mlx.nn.selu — MLX 0.3.0 documentation
+ mlx.nn.selu — MLX 0.5.0 documentation
@@ -47,8 +47,8 @@
mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers where \(\lambda = 1.0507\) and \(\alpha = 1.67326\) .
-See also elu()
+See also elu()
@@ -718,20 +736,20 @@ document.write(`
previous
-
mlx.nn.relu
+
mlx.nn.relu6
next
-
mlx.nn.softshrink
+
mlx.nn.sigmoid
diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.sigmoid.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.sigmoid.html
new file mode 100644
index 00000000..a883f224
--- /dev/null
+++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.sigmoid.html
@@ -0,0 +1,807 @@
+
+
+
+
+
+
+
+
+
+
+
+
mlx.nn.sigmoid — MLX 0.5.0 documentation
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
Skip to main content
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+mlx.nn.sigmoid
+
+
+mlx.nn. sigmoid ( x )
+Applies the sigmoid function.
+
+\[\text{Sigmoid}(x) = \sigma(x) = \frac{1}{1 + \exp(-x)}\]
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
\ No newline at end of file
diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.silu.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.silu.html
index 7481cae5..7184aa9f 100644
--- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.silu.html
+++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.silu.html
@@ -9,7 +9,7 @@
-
mlx.nn.silu — MLX 0.3.0 documentation
+
mlx.nn.silu — MLX 0.5.0 documentation
@@ -47,8 +47,8 @@
-
-
+
+
@@ -135,8 +135,8 @@
-
-
+
+
@@ -234,6 +234,9 @@
mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers
previous
-
mlx.nn.softshrink
+
mlx.nn.sigmoid
next
-
mlx.nn.step
+
mlx.nn.softmax
diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.softmax.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.softmax.html
new file mode 100644
index 00000000..d8ffc8ef
--- /dev/null
+++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.softmax.html
@@ -0,0 +1,806 @@
+
+
+
+
+
+
+
+
+
+
+
+
mlx.nn.softmax — MLX 0.5.0 documentation
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
Skip to main content
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+mlx.nn.softmax
+
+
+mlx.nn. softmax ( x axis = - 1 )
+Applies the Softmax function.
+Applies \(\frac{e^{x_i}}{\sum_j e^{x_j}}\) element wise.
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
\ No newline at end of file
diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.softplus.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.softplus.html
new file mode 100644
index 00000000..2cd2e7c5
--- /dev/null
+++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.softplus.html
@@ -0,0 +1,806 @@
+
+
+
+
+
+
+
+
+
+
+
+
mlx.nn.softplus — MLX 0.5.0 documentation
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
Skip to main content
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+mlx.nn.softplus
+
+
+mlx.nn. softplus ( x )
+Applies the Softplus function.
+Applies \(\log(1 + \exp(x))\) element wise.
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
\ No newline at end of file
diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.softshrink.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.softshrink.html
index cd215037..b6f16ec6 100644
--- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.softshrink.html
+++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.softshrink.html
@@ -9,7 +9,7 @@
-
mlx.nn.softshrink — MLX 0.3.0 documentation
+
mlx.nn.softshrink — MLX 0.5.0 documentation
@@ -47,8 +47,8 @@
-
-
+
+
@@ -135,8 +135,8 @@
-
-
+
+
@@ -234,6 +234,9 @@
mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers
previous
-
mlx.nn.selu
+
mlx.nn.softplus
next
-
mlx.nn.silu
+
mlx.nn.step
diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.step.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.step.html
index 520f03b8..a8cec871 100644
--- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.step.html
+++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.step.html
@@ -9,7 +9,7 @@
-
mlx.nn.step — MLX 0.3.0 documentation
+
mlx.nn.step — MLX 0.5.0 documentation
@@ -47,8 +47,8 @@
-
-
+
+
@@ -135,8 +135,8 @@
-
-
+
+
@@ -234,6 +234,9 @@
mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers
previous
-
mlx.nn.silu
+
mlx.nn.softshrink
next
-
Loss Functions
+
mlx.nn.tanh
diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.tanh.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.tanh.html
new file mode 100644
index 00000000..47a4fe2f
--- /dev/null
+++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.tanh.html
@@ -0,0 +1,805 @@
+
+
+
+
+
+
+
+
+
+
+
+
mlx.nn.tanh — MLX 0.5.0 documentation
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
Skip to main content
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+mlx.nn.tanh
+
+
+mlx.nn. tanh ( x )
+Applies the hyperbolic tangent function.
+Simply mx.tanh(x)
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
\ No newline at end of file
diff --git a/docs/build/html/python/nn/functions.html b/docs/build/html/python/nn/functions.html
index 5b69c72e..63c44533 100644
--- a/docs/build/html/python/nn/functions.html
+++ b/docs/build/html/python/nn/functions.html
@@ -9,7 +9,7 @@
-
Functions — MLX 0.3.0 documentation
+
Functions — MLX 0.5.0 documentation
@@ -46,8 +46,8 @@
-
-
+
+
@@ -134,8 +134,8 @@
-
-
+
+
@@ -233,6 +233,9 @@
mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers
-gelu
elu
Applies the Exponential Linear Unit.
gelu
Applies the Gaussian Error Linear Units function.
gelu_approx
gelu_approx
An approximation to Gaussian Error Linear Unit.
gelu_fast_approx
gelu_fast_approx
A fast approximation to Gaussian Error Linear Unit.
glu
Applies the gated linear unit function.
hardswish
Applies the hardswish function, element-wise.
leaky_relu
Applies the Leaky Rectified Linear Unit.
log_sigmoid
Applies the Log Sigmoid function.
log_softmax
Applies the Log Softmax function.
mish
Applies the Mish function, element-wise.
relu
Applies the Rectified Linear Unit.
selu
relu6
Applies the Rectified Linear Unit 6.
selu
Applies the Scaled Exponential Linear Unit.
softshrink
Applies the Softshrink activation function.
sigmoid
Applies the sigmoid function.
silu
silu
Applies the Sigmoid Linear Unit.
softmax
Applies the Softmax function.
softplus
Applies the Softplus function.
softshrink
Applies the Softshrink activation function.
step
Applies the Step Activation Function.
tanh
Applies the hyperbolic tangent function.
@@ -740,20 +791,20 @@ simple functions.
previous
-
mlx.nn.Transformer
+
mlx.nn.Upsample
next
-
mlx.nn.gelu
+
mlx.nn.elu
diff --git a/docs/build/html/python/nn/init.html b/docs/build/html/python/nn/init.html
index a1fbc557..ec37f85b 100644
--- a/docs/build/html/python/nn/init.html
+++ b/docs/build/html/python/nn/init.html
@@ -9,7 +9,7 @@
-
Initializers — MLX 0.3.0 documentation
+
Initializers — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
-
-
+
+
@@ -233,6 +233,9 @@
mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers Layers — MLX 0.3.0 documentation
+ Layers — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers Transformer
Implements a standard Transformer model.
Upsample
Upsample the input signal spatially.
diff --git a/docs/build/html/python/nn/losses.html b/docs/build/html/python/nn/losses.html
index 25052a2f..c190a57b 100644
--- a/docs/build/html/python/nn/losses.html
+++ b/docs/build/html/python/nn/losses.html
@@ -9,7 +9,7 @@
Loss Functions — MLX 0.3.0 documentation
+ Loss Functions — MLX 0.5.0 documentation
@@ -48,7 +48,7 @@
mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample
Functions Loss Functions Schedulers mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers Operations — MLX 0.3.0 documentation
+ Operations — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers array_equal
Array equality check.
broadcast_to
atleast_1d
Convert all arrays to have at least one dimension.
atleast_2d
Convert all arrays to have at least two dimensions.
atleast_3d
Convert all arrays to have at least three dimensions.
broadcast_to
Broadcast an array to the given shape.
ceil
ceil
Element-wise ceil.
clip
clip
Clip the values of the array between the given minimum and maximum.
concatenate
concatenate
Concatenate the arrays along the given axis.
convolve
convolve
The discrete convolution of 1D arrays.
conv1d
conv1d
1D convolution over an input with several channels
conv2d
conv2d
2D convolution over an input with several channels
conv_general
General convolution over an input with several channels
cos
Element-wise cosine.
Optimizers — MLX 0.3.0 documentation
+ Optimizers — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers optimizer state .
Schedulers
mlx.optimizers.AdaDelta — MLX 0.3.0 documentation
+ mlx.optimizers.AdaDelta — MLX 0.5.0 documentation
@@ -135,8 +135,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.optimizers.Adafactor — MLX 0.3.0 documentation
+ mlx.optimizers.Adafactor — MLX 0.5.0 documentation
@@ -135,8 +135,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.optimizers.Adagrad — MLX 0.3.0 documentation
+ mlx.optimizers.Adagrad — MLX 0.5.0 documentation
@@ -135,8 +135,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.optimizers.Adam — MLX 0.3.0 documentation
+ mlx.optimizers.Adam — MLX 0.5.0 documentation
@@ -135,8 +135,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.optimizers.AdamW — MLX 0.3.0 documentation
+ mlx.optimizers.AdamW — MLX 0.5.0 documentation
@@ -135,8 +135,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.optimizers.Adamax — MLX 0.3.0 documentation
+ mlx.optimizers.Adamax — MLX 0.5.0 documentation
@@ -135,8 +135,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.optimizers.Lion — MLX 0.3.0 documentation
+ mlx.optimizers.Lion — MLX 0.5.0 documentation
@@ -135,8 +135,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.optimizers.Optimizer.apply_gradients — MLX 0.3.0 documentation
+ mlx.optimizers.Optimizer.apply_gradients — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.optimizers.Optimizer.init — MLX 0.3.0 documentation
+ mlx.optimizers.Optimizer.init — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.optimizers.Optimizer.state — MLX 0.3.0 documentation
+ mlx.optimizers.Optimizer.state — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.optimizers.Optimizer.update — MLX 0.3.0 documentation
+ mlx.optimizers.Optimizer.update — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.optimizers.RMSprop — MLX 0.3.0 documentation
+ mlx.optimizers.RMSprop — MLX 0.5.0 documentation
@@ -135,8 +135,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.optimizers.SGD — MLX 0.3.0 documentation
+ mlx.optimizers.SGD — MLX 0.5.0 documentation
@@ -135,8 +135,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.optimizers.cosine_decay — MLX 0.3.0 documentation
+ mlx.optimizers.cosine_decay — MLX 0.5.0 documentation
@@ -46,8 +46,8 @@
mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.optimizers.cosine_decay
-mlx.optimizers. cosine_decay ( init : float decay_steps : int )
+mlx.optimizers. cosine_decay ( init : float decay_steps : int ) → Callable Make a cosine decay scheduler.
Parameters:
@@ -731,20 +749,20 @@ value is constant for steps beyond
previous
-
mlx.optimizers.exponential_decay
+
Schedulers
next
-
Tree Utils
+
mlx.optimizers.exponential_decay
diff --git a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.exponential_decay.html b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.exponential_decay.html
index e44d27e1..8df8aa71 100644
--- a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.exponential_decay.html
+++ b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.exponential_decay.html
@@ -9,7 +9,7 @@
mlx.optimizers.exponential_decay — MLX 0.3.0 documentation
+ mlx.optimizers.exponential_decay — MLX 0.5.0 documentation
@@ -46,8 +46,8 @@
mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.optimizers.exponential_decay
-mlx.optimizers. exponential_decay ( init : float decay_rate : float )
+mlx.optimizers. exponential_decay ( init : float decay_rate : float ) → Callable Make an exponential decay scheduler.
Parameters:
@@ -730,20 +748,20 @@ document.write(`
previous
-
mlx.optimizers.step_decay
+
mlx.optimizers.cosine_decay
next
-
mlx.optimizers.cosine_decay
+
mlx.optimizers.join_schedules
diff --git a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.join_schedules.html b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.join_schedules.html
new file mode 100644
index 00000000..fdef9026
--- /dev/null
+++ b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.join_schedules.html
@@ -0,0 +1,829 @@
+
+
+
+
+
+
+
+
+
+
+
+
mlx.optimizers.join_schedules — MLX 0.5.0 documentation
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
Skip to main content
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
mlx.optimizers.join_schedules
+
+
+
+
+
+
+
+
+
+
+mlx.optimizers.join_schedules
+
+
+mlx.optimizers. join_schedules ( schedules : List [ Callable ] boundaries : List [ int ] ) → Callable
+Join multiple schedules to create a new schedule.
+
+Parameters:
+
+schedules (list ( Callable ) ) – A list of schedules. Schedule \(i+1\)
+receives a step count indicating the number of steps since
+the \(i\) -th boundary.
boundaries (list ( int ) ) – A list of integers of length len(schedules) - 1
+
+Example
+>>> warmup = optim . linear_schedule ( 0 , 1e-1 , steps = 10 )
+>>> cosine = optim . cosine_decay ( 1e-1 , 200 )
+>>> lr_schedule = optim . join_schedules ([ warmup , cosine ], [ 10 ])
+>>> optimizer = optim . Adam ( learning_rate = lr_schedule )
+>>> optimizer . learning_rate
+array(0.0, dtype=float32)
+>>> for _ in range ( 12 ): optimizer . update ({}, {})
+...
+>>> optimizer . learning_rate
+array(0.0999938, dtype=float32)
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
\ No newline at end of file
diff --git a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.linear_schedule.html b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.linear_schedule.html
new file mode 100644
index 00000000..07d8eae5
--- /dev/null
+++ b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.linear_schedule.html
@@ -0,0 +1,825 @@
+
+
+
+
+
+
+
+
+
+
+
+
mlx.optimizers.linear_schedule — MLX 0.5.0 documentation
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
Skip to main content
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
mlx.optimizers.linear_schedule
+
+
+
+
+
+
+
+
+
+
+mlx.optimizers.linear_schedule
+
+
+mlx.optimizers. linear_schedule ( init : float end : float steps : int ) → Callable
+Make a linear scheduler.
+
+Parameters:
+
+init (float
end (float
steps (int end steps
+
+Example
+>>> warmup = optim . linear_schedule ( 0 , 1e-1 , 100 )
+>>> optimizer = optim . Adam ( learning_rate = warmup )
+>>> optimizer . learning_rate
+array(0.0, dtype=float32)
+>>> for _ in range ( 101 ): optimizer . update ({}, {})
+...
+>>> optimizer . learning_rate
+array(0.1, dtype=float32)
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
\ No newline at end of file
diff --git a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.step_decay.html b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.step_decay.html
index 65995904..2d21c322 100644
--- a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.step_decay.html
+++ b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.step_decay.html
@@ -9,7 +9,7 @@
-
mlx.optimizers.step_decay — MLX 0.3.0 documentation
+
mlx.optimizers.step_decay — MLX 0.5.0 documentation
@@ -46,8 +46,8 @@
-
-
+
+
@@ -134,8 +134,8 @@
-
-
+
+
@@ -233,6 +233,9 @@
mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers mlx.optimizers.step_decay
-mlx.optimizers. step_decay ( init : float decay_rate : float step_size : int )
+mlx.optimizers. step_decay ( init : float decay_rate : float step_size : int ) → Callable Make a step decay scheduler.
Parameters:
@@ -731,20 +749,20 @@ document.write(`
previous
-
Schedulers
+
mlx.optimizers.linear_schedule
next
-
mlx.optimizers.exponential_decay
+
Tree Utils
diff --git a/docs/build/html/python/optimizers/common_optimizers.html b/docs/build/html/python/optimizers/common_optimizers.html
index 82c42a40..160df84a 100644
--- a/docs/build/html/python/optimizers/common_optimizers.html
+++ b/docs/build/html/python/optimizers/common_optimizers.html
@@ -9,7 +9,7 @@
-
Common Optimizers — MLX 0.3.0 documentation
+
Common Optimizers — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
-
-
+
+
@@ -233,6 +233,9 @@
mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers Optimizer — MLX 0.3.0 documentation
+ Optimizer — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers Schedulers — MLX 0.3.0 documentation
+ Schedulers — MLX 0.5.0 documentation
@@ -46,7 +46,7 @@
mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers Schedulers
-step_decay
Make a step decay scheduler.
cosine_decay
Make a cosine decay scheduler.
exponential_decay
Make an exponential decay scheduler.
cosine_decay
Make a cosine decay scheduler.
join_schedules
Join multiple schedules to create a new schedule.
linear_schedule
Make a linear scheduler.
step_decay
Make a step decay scheduler.
@@ -726,11 +750,11 @@ document.write(`
next
-
mlx.optimizers.step_decay
+
mlx.optimizers.cosine_decay
diff --git a/docs/build/html/python/random.html b/docs/build/html/python/random.html
index 320d519e..a7c3e1bb 100644
--- a/docs/build/html/python/random.html
+++ b/docs/build/html/python/random.html
@@ -9,7 +9,7 @@
Random — MLX 0.3.0 documentation
+ Random — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers Transforms — MLX 0.3.0 documentation
+ Transforms — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers Tree Utils — MLX 0.3.0 documentation
+ Tree Utils — MLX 0.5.0 documentation
@@ -47,7 +47,7 @@
mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers
previous
-
mlx.optimizers.cosine_decay
+
mlx.optimizers.step_decay
Search - MLX 0.3.0 documentation
+ Search - MLX 0.5.0 documentation
@@ -133,8 +133,8 @@
-
@@ -232,6 +232,9 @@
mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers Compilation — MLX 0.3.0 documentation
+ Compilation — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers Function Transforms — MLX 0.3.0 documentation
+ Function Transforms — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers Indexing Arrays — MLX 0.3.0 documentation
+ Indexing Arrays — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers Lazy Evaluation — MLX 0.3.0 documentation
+ Lazy Evaluation — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers Conversion to NumPy and Other Frameworks — MLX 0.3.0 documentation
+ Conversion to NumPy and Other Frameworks — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers Quick Start Guide — MLX 0.3.0 documentation
+ Quick Start Guide — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers Saving and Loading Arrays — MLX 0.3.0 documentation
+ Saving and Loading Arrays — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers Unified Memory — MLX 0.3.0 documentation
+ Unified Memory — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers Using Streams — MLX 0.3.0 documentation
+ Using Streams — MLX 0.5.0 documentation
@@ -134,8 +134,8 @@
- mlx.core.argpartition mlx.core.argsort mlx.core.array_equal mlx.core.atleast_1d mlx.core.atleast_2d mlx.core.atleast_3d mlx.core.broadcast_to mlx.core.ceil mlx.core.clip mlx.core.convolve mlx.core.conv1d mlx.core.conv2d mlx.core.conv_general mlx.core.cos mlx.core.cosh mlx.core.dequantize mlx.nn.Softshrink mlx.nn.Step mlx.nn.Transformer mlx.nn.Upsample Functions Loss Functions Schedulers
-
+ 
-
+ 
-
+ 
-
+