+ power() (in module mlx.core)
+PReLU (class in mlx.nn)
prelu (class in mlx.nn)
+prod (C++ function) , [1] , [2] , [3]
prod() (array method)
@@ -1558,6 +1824,8 @@ document.write(`
+ quantized_matmul (C++ function)
+quantized_matmul() (in module mlx.core)
QuantizedEmbedding (class in mlx.nn)
@@ -1579,9 +1849,13 @@ document.write(`
R
+ radians (C++ function)
+radians() (in module mlx.core)
randint() (in module mlx.core.random)
+reciprocal (C++ function)
reciprocal() (array method)
@@ -1596,12 +1870,18 @@ document.write(`
ReLU6 (class in mlx.nn)
relu6 (class in mlx.nn)
+remainder (C++ function)
remainder() (in module mlx.core)
+repeat (C++ function) , [1]
repeat() (in module mlx.core)
reset_peak_memory() (in module mlx.core.metal)
+reshape (C++ function)
reshape() (array method)
@@ -1616,6 +1896,8 @@ document.write(`
rfft2() (in module mlx.core.fft)
rfftn() (in module mlx.core.fft)
+right_shift (C++ function)
right_shift() (in module mlx.core)
RoPE (class in mlx.nn)
rope() (in module mlx.core.fast)
+round (C++ function) , [1]
round() (array method)
@@ -1637,6 +1921,8 @@ document.write(`
(in module mlx.core)
+ rsqrt (C++ function)
+rsqrt() (array method)
shape (array property)
sigmoid (class in mlx.nn)
+ sigmoid (C++ function)
+
+ sigmoid() (in module mlx.core)
+sign (C++ function)
sign() (in module mlx.core)
SiLU (class in mlx.nn)
silu (class in mlx.nn)
+sin (C++ function)
sin() (array method)
@@ -1701,13 +2005,21 @@ document.write(`
(in module mlx.core)
sinh (C++ function)
+sinh() (in module mlx.core)
SinusoidalPositionalEncoding (class in mlx.nn)
size (array property)
+slice (C++ function) , [1]
+slice_update (C++ function) , [1]
smooth_l1_loss (class in mlx.nn.losses)
+softmax (C++ function) , [1] , [2]
Softmax (class in mlx.nn)
softmax() (in module mlx.core)
Softmin (class in mlx.nn)
-softmin (class in mlx.nn)
+ sqrt (C++ function)
+sqrt() (array method)
square (C++ function)
+square() (array method)
squeeze (C++ function) , [1] , [2]
+squeeze() (array method)
stack (C++ function) , [1]
+stack() (in module mlx.core)
start_capture() (in module mlx.core.metal)
@@ -1769,6 +2093,8 @@ document.write(`
(Optimizer property)
std (C++ function) , [1] , [2] , [3]
+std() (in module mlx.core)
Step (class in mlx.nn)
@@ -1778,14 +2104,20 @@ document.write(`
step_decay() (in module mlx.optimizers)
stop_capture() (in module mlx.core.metal)
+stop_gradient (C++ function)
stop_gradient() (in module mlx.core)
Stream (class in mlx.core)
stream() (in module mlx.core)
+subtract (C++ function)
subtract() (in module mlx.core)
+sum (C++ function) , [1] , [2] , [3]
sum() (array method)
@@ -1794,6 +2126,8 @@ document.write(`
svd() (in module mlx.core.linalg)
+swapaxes (C++ function)
swapaxes() (array method)
@@ -1810,38 +2144,56 @@ document.write(`
+ var (C++ function) , [1] , [2] , [3]
+var() (array method)
@@ -1915,6 +2275,8 @@ document.write(`
array mlx::core::flatten (const array &a, StreamOrDevice s={}) Flatten the array to 1D. array mlx::core::hadamard_transform (const array &a, float scale=1.0f, StreamOrDevice s={}) Multiply the array by the Hadamard matrix of corresponding size. array mlx::core::hadamard_transform (const array &a, std::optional< float > scale=std::nullopt, StreamOrDevice s={}) Multiply the array by the Hadamard matrix of corresponding size. array mlx::core::squeeze (const array &a, const std::vector< int > &axes, StreamOrDevice s={}) Remove singleton dimensions at the given axes. array mlx::core::where (const array &condition, const array &x, const array &y, StreamOrDevice s={}) Select from x or y depending on condition. array mlx::core::nan_to_num (const array &a, float nan=0.0f, const std::optional< float > &posinf=std::nullopt, const std::optional< float > &neginf=std::nullopt, StreamOrDevice s={}) Replace NaN and infinities with finite numbers. array mlx::core::all (const array &a, bool keepdims, StreamOrDevice s={}) True if all elements in the array are true (or non-zero). 137 std::optional<float> scale = std::nullopt,
-
-
-
+
+
+
+
413 const std::optional<float>& posinf = std::nullopt,
+
414 const std::optional<float>& neginf = std::nullopt,
+
+
+
+
-
-
-
-
-
-
-
421 bool equal_nan =
false ,
-
-
-
-
-
-
-
-
431 bool equal_nan =
false ,
-
-
-
-
-
440 const std::vector<int>& axes,
-
441 bool keepdims =
false ,
-
-
-
-
-
-
451 bool keepdims =
false ,
-
-
-
-
-
-
-
+
+
+
+
+
+
+
429 bool equal_nan =
false ,
+
+
+
+
+
+
+
+
439 bool equal_nan =
false ,
+
+
+
+
+
448 const std::vector<int>& axes,
+
449 bool keepdims =
false ,
+
+
+
+
+
+
459 bool keepdims =
false ,
+
+
+
+
-
-
-
-
466 const std::vector<int>& axes,
-
467 bool keepdims =
false ,
-
-
-
-
-
-
477 bool keepdims =
false ,
-
-
-
-
-
-
-
+
+
+
+
474 const std::vector<int>& axes,
+
475 bool keepdims =
false ,
+
+
+
+
+
+
485 bool keepdims =
false ,
+
+
+
+
-
-
-
-
489 const std::vector<int>& axes,
-
490 bool keepdims =
false ,
-
-
-
-
-
-
497 bool keepdims =
false ,
-
-
-
-
-
-
-
+
+
+
+
497 const std::vector<int>& axes,
+
498 bool keepdims =
false ,
+
+
+
+
+
+
505 bool keepdims =
false ,
+
+
+
+
-
-
-
-
509 const std::vector<int>& axes,
-
510 bool keepdims =
false ,
-
-
-
-
-
-
517 bool keepdims =
false ,
-
-
-
-
-
-
-
+
+
+
+
517 const std::vector<int>& axes,
+
518 bool keepdims =
false ,
+
+
+
+
+
+
525 bool keepdims =
false ,
+
+
+
+
-
-
-
-
530 const std::vector<int>& axes,
-
531 bool keepdims =
false ,
-
-
-
-
-
-
-
540 bool keepdims =
false ,
-
-
-
-
-
-
-
-
+
+
+
+
538 const std::vector<int>& axes,
+
539 bool keepdims =
false ,
+
+
+
+
+
+
+
548 bool keepdims =
false ,
+
+
+
+
+
-
-
-
-
554 const std::vector<int>& axes,
-
555 bool keepdims =
false ,
-
-
-
-
-
-
-
564 bool keepdims =
false ,
-
-
-
-
-
-
-
-
+
+
+
+
562 const std::vector<int>& axes,
+
563 bool keepdims =
false ,
+
+
+
+
+
+
+
572 bool keepdims =
false ,
+
+
+
+
+
-
-
-
-
577 const std::vector<int>& axes,
-
578 bool keepdims =
false ,
-
-
-
-
-
-
585 bool keepdims =
false ,
-
-
-
-
-
-
-
+
+
+
+
585 const std::vector<int>& axes,
+
586 bool keepdims =
false ,
+
+
+
+
+
+
593 bool keepdims =
false ,
+
+
+
+
-
-
-
-
597 const std::vector<int>& axes,
-
598 bool keepdims =
false ,
-
-
-
-
-
-
605 bool keepdims =
false ,
-
-
-
-
-
-
-
+
+
+
+
605 const std::vector<int>& axes,
+
606 bool keepdims =
false ,
+
+
+
+
+
+
613 bool keepdims =
false ,
+
+
+
+
-
-
-
-
617 const std::vector<int>& axes,
-
618 bool keepdims =
false ,
-
-
-
-
-
-
625 bool keepdims =
false ,
-
-
-
-
-
-
631 return argmin (a,
false , s);
-
+
+
+
+
625 const std::vector<int>& axes,
+
626 bool keepdims =
false ,
+
+
+
+
+
+
633 bool keepdims =
false ,
+
+
+
+
+
+
639 return argmin (a,
false , s);
+
-
-
-
-
-
638 bool keepdims =
false ,
-
-
-
-
-
-
644 return argmax (a,
false , s);
-
+
+
+
+
+
646 bool keepdims =
false ,
+
+
+
+
+
+
652 return argmax (a,
false , s);
+
-
-
-
-
-
651 bool keepdims =
false ,
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
+
+
+
+
+
659 bool keepdims =
false ,
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
-
-
-
-
705 const std::vector<int>& axes,
-
706 bool keepdims =
false ,
-
-
-
-
-
-
713 bool keepdims =
false ,
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
+
+
+
+
713 const std::vector<int>& axes,
+
714 bool keepdims =
false ,
+
+
+
+
+
+
721 bool keepdims =
false ,
+
+
+
+
+
+
+
+
+
+
-
-
-
-
-
-
-
-
-
+
+
+
+
+
+
+
+
+
+
+
+
-
-
-
-
-
-
-
-
-
-
+
+
+
+
+
-
-
-
-
-
-
-
-
+
+
+
+
+
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
+
+
-
-
-
-
-
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
897 return round (a, 0, s);
-
+
+
-
-
-
-
-
-
906 const std::vector<array>& indices,
-
907 const std::vector<int>& axes,
-
908 const std::vector<int>& slice_sizes,
-
-
-
-
-
912 const array & indices,
-
-
914 const std::vector<int>& slice_sizes,
-
-
916 return gather (a, {indices}, std::vector<int>{axis}, slice_sizes, s);
-
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
905 return round (a, 0, s);
+
-
-
-
-
922 const array & indices,
-
-
-
-
-
-
-
-
932 const array & indices,
-
-
-
-
-
-
1036 const std::vector<array>& indices,
-
1037 const array & updates,
-
1038 const std::vector<int>& axes,
-
-
-
-
-
1042 const array & indices,
-
1043 const array & updates,
-
-
-
1046 return scatter (a, {indices}, updates, std::vector<int>{axis}, s);
-
+
+
+
+
+
+
914 const std::vector<array>& indices,
+
915 const std::vector<int>& axes,
+
916 const std::vector<int>& slice_sizes,
+
+
+
+
+
920 const array & indices,
+
+
922 const std::vector<int>& slice_sizes,
+
+
924 return gather (a, {indices}, std::vector<int>{axis}, slice_sizes, s);
+
-
-
-
-
1052 const std::vector<array>& indices,
-
1053 const array & updates,
-
1054 const std::vector<int>& axes,
-
-
-
-
-
1058 const array & indices,
-
1059 const array & updates,
-
-
-
1062 return scatter_add (a, {indices}, updates, std::vector<int>{axis}, s);
-
+
+
+
+
930 const array & indices,
+
+
+
+
+
+
+
+
940 const array & indices,
+
+
+
+
+
+
1044 const std::vector<array>& indices,
+
1045 const array & updates,
+
1046 const std::vector<int>& axes,
+
+
+
+
+
1050 const array & indices,
+
1051 const array & updates,
+
+
+
1054 return scatter (a, {indices}, updates, std::vector<int>{axis}, s);
+
-
-
-
-
1068 const std::vector<array>& indices,
-
1069 const array & updates,
-
1070 const std::vector<int>& axes,
-
-
-
-
-
1074 const array & indices,
-
1075 const array & updates,
-
-
-
1078 return scatter_prod (a, {indices}, updates, std::vector<int>{axis}, s);
-
+
+
+
+
1060 const std::vector<array>& indices,
+
1061 const array & updates,
+
1062 const std::vector<int>& axes,
+
+
+
+
+
1066 const array & indices,
+
1067 const array & updates,
+
+
+
1070 return scatter_add (a, {indices}, updates, std::vector<int>{axis}, s);
+
-
-
-
-
1084 const std::vector<array>& indices,
-
1085 const array & updates,
-
1086 const std::vector<int>& axes,
-
-
-
-
-
1090 const array & indices,
-
1091 const array & updates,
-
-
-
1094 return scatter_max (a, {indices}, updates, std::vector<int>{axis}, s);
-
+
+
+
+
1076 const std::vector<array>& indices,
+
1077 const array & updates,
+
1078 const std::vector<int>& axes,
+
+
+
+
+
1082 const array & indices,
+
1083 const array & updates,
+
+
+
1086 return scatter_prod (a, {indices}, updates, std::vector<int>{axis}, s);
+
-
-
-
1099 const std::vector<array>& indices,
-
1100 const array & updates,
-
1101 const std::vector<int>& axes,
-
-
-
-
-
1105 const array & indices,
-
1106 const array & updates,
-
-
-
1109 return scatter_min (a, {indices}, updates, std::vector<int>{axis}, s);
-
+
+
+
+
1092 const std::vector<array>& indices,
+
1093 const array & updates,
+
1094 const std::vector<int>& axes,
+
+
+
+
+
1098 const array & indices,
+
1099 const array & updates,
+
+
+
1102 return scatter_max (a, {indices}, updates, std::vector<int>{axis}, s);
+
-
-
-
-
-
-
-
-
1121 const std::vector<int>& axes,
-
1122 bool precise =
false ,
-
-
-
-
-
-
-
-
1131 return softmax (a, std::vector<int>{axis}, precise, s);
-
+
+
+
1107 const std::vector<array>& indices,
+
1108 const array & updates,
+
1109 const std::vector<int>& axes,
+
+
+
+
+
1113 const array & indices,
+
1114 const array & updates,
+
+
+
1117 return scatter_min (a, {indices}, updates, std::vector<int>{axis}, s);
+
-
-
-
-
-
-
-
1141 bool reverse =
false ,
-
1142 bool inclusive =
true ,
-
+
+
+
+
+
+
+
+
1129 const std::vector<int>& axes,
+
1130 bool precise =
false ,
+
+
+
+
+
+
+
+
1139 return softmax (a, std::vector<int>{axis}, precise, s);
+
+
+
+
-
+
1149 bool reverse =
false ,
1150 bool inclusive =
true ,
-
+
1157 bool reverse =
false ,
1158 bool inclusive =
true ,
-
+
1165 bool reverse =
false ,
1166 bool inclusive =
true ,
-
-
-
-
1173 std::vector<int> stride = {},
-
1174 std::vector<int> padding_lo = {},
-
1175 std::vector<int> padding_hi = {},
-
1176 std::vector<int> kernel_dilation = {},
-
1177 std::vector<int> input_dilation = {},
-
-
-
-
-
-
-
-
1185 const array & weight,
-
1186 std::vector<int> stride = {},
-
1187 std::vector<int> padding = {},
-
1188 std::vector<int> kernel_dilation = {},
-
1189 std::vector<int> input_dilation = {},
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
+
+
+
+
1173 bool reverse =
false ,
+
1174 bool inclusive =
true ,
+
+
+
+
+
+
1181 std::vector<int> stride = {},
+
1182 std::vector<int> padding_lo = {},
+
1183 std::vector<int> padding_hi = {},
+
1184 std::vector<int> kernel_dilation = {},
+
1185 std::vector<int> input_dilation = {},
+
+
+
+
+
+
+
+
1193 const array & weight,
+
1194 std::vector<int> stride = {},
+
1195 std::vector<int> padding = {},
+
1196 std::vector<int> kernel_dilation = {},
+
1197 std::vector<int> input_dilation = {},
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
-
-
-
-
1209 const array & weight,
-
-
-
-
-
-
-
-
-
1219 const array & weight,
-
1220 const std::pair<int, int>& stride = {1, 1},
-
1221 const std::pair<int, int>& padding = {0, 0},
-
1222 const std::pair<int, int>& dilation = {1, 1},
-
-
-
-
-
-
1229 const array & weight,
-
1230 const std::tuple<int, int, int>& stride = {1, 1, 1},
-
1231 const std::tuple<int, int, int>& padding = {0, 0, 0},
-
1232 const std::tuple<int, int, int>& dilation = {1, 1, 1},
-
-
-
-
-
-
-
1240 const array & scales,
-
1241 const array & biases,
-
-
1243 int group_size = 64,
-
-
-
-
-
-
1250 int group_size = 64,
-
-
-
-
-
-
1257 const array & scales,
-
1258 const array & biases,
-
1259 int group_size = 64,
-
-
-
-
-
-
-
1267 const array & scales,
-
1268 const array & biases,
-
1269 std::optional<array> lhs_indices = std::nullopt,
-
1270 std::optional<array> rhs_indices = std::nullopt,
-
-
1272 int group_size = 64,
-
-
-
-
-
-
-
-
-
-
-
-
-
1286 const std::vector<int>& axes_a,
-
1287 const std::vector<int>& axes_b,
-
-
-
-
-
-
-
-
-
-
-
1301 const float & alpha = 1.f,
-
1302 const float & beta = 1.f,
-
-
-
+
+
+
+
1217 const array & weight,
+
+
+
+
+
+
+
+
+
1227 const array & weight,
+
1228 const std::pair<int, int>& stride = {1, 1},
+
1229 const std::pair<int, int>& padding = {0, 0},
+
1230 const std::pair<int, int>& dilation = {1, 1},
+
+
+
+
+
+
1237 const array & weight,
+
1238 const std::tuple<int, int, int>& stride = {1, 1, 1},
+
1239 const std::tuple<int, int, int>& padding = {0, 0, 0},
+
1240 const std::tuple<int, int, int>& dilation = {1, 1, 1},
+
+
+
+
+
+
+
1248 const array & scales,
+
1249 const array & biases,
+
+
1251 int group_size = 64,
+
+
+
+
+
+
1258 int group_size = 64,
+
+
+
+
+
+
1265 const array & scales,
+
1266 const array & biases,
+
1267 int group_size = 64,
+
+
+
+
+
+
+
1275 const array & scales,
+
1276 const array & biases,
+
1277 std::optional<array> lhs_indices = std::nullopt,
+
1278 std::optional<array> rhs_indices = std::nullopt,
+
+
1280 int group_size = 64,
+
+
+
+
+
+
+
+
+
+
+
+
+
1294 const std::vector<int>& axes_a,
+
1295 const std::vector<int>& axes_b,
+
+
+
+
+
+
+
+
-
-
1310 std::optional<array> mask_out = std::nullopt,
-
1311 std::optional<array> mask_lhs = std::nullopt,
-
1312 std::optional<array> mask_rhs = std::nullopt,
-
-
-
-
-
-
1319 std::optional<array> lhs_indices = std::nullopt,
-
1320 std::optional<array> rhs_indices = std::nullopt,
+
1309 const float & alpha = 1.f,
+
1310 const float & beta = 1.f,
+
+
+
+
+
+
+
1318 std::optional<array> mask_out = std::nullopt,
+
1319 std::optional<array> mask_lhs = std::nullopt,
+
1320 std::optional<array> mask_rhs = std::nullopt,
-
-
-
-
-
+
+
+
+
1327 std::optional<array> lhs_indices = std::nullopt,
+
1328 std::optional<array> rhs_indices = std::nullopt,
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
-
1356 const std::vector<array>& inputs,
-
1357 const std::vector<array>& dependencies);
-
-
-
-
1362 const std::vector<array>& a,
-
-
-
-
1366 const std::vector<array>& a,
-
-
-
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
1364 const std::vector<array>& inputs,
+
1365 const std::vector<array>& dependencies);
+
+
+
1370 const std::vector<array>& a,
-
-
-
-
1379 std::vector<int> axes,
-
-
-
-
-
-
-
-
-
-
-
+
+
+
1374 const std::vector<array>& a,
+
+
+
+
1378 const std::vector<array>& a,
+
+
+
+
+
1387 std::vector<int> axes,
+
+
+
+
+
-
-
+
+
-
-
+
+
-
-
+
+
-
-
+
+
+
+
+
+
+
+
-
array hadamard_transform(const array &a, float scale=1.0f, StreamOrDevice s={})
Multiply the array by the Hadamard matrix of corresponding size.
array scatter_max(const array &a, const std::vector< array > &indices, const array &updates, const std::vector< int > &axes, StreamOrDevice s={})
Scatter and max updates to given linear indices.
array floor_divide(const array &a, const array &b, StreamOrDevice s={})
Compute integer division.
array radians(const array &a, StreamOrDevice s={})
Convert the elements of an array from Degrees to Radians.
@@ -1345,6 +1351,7 @@ $(function() { codefold.init(0); });
array reciprocal(const array &a, StreamOrDevice s={})
The reciprocal (1/x) of the elements in an array.
array tri(int n, int m, int k, Dtype type, StreamOrDevice s={})
array pad(const array &a, const std::vector< int > &axes, const std::vector< int > &low_pad_size, const std::vector< int > &high_pad_size, const array &pad_value=array(0), StreamOrDevice s={})
Pad an array with a constant value.
+
array nan_to_num(const array &a, float nan=0.0f, const std::optional< float > &posinf=std::nullopt, const std::optional< float > &neginf=std::nullopt, StreamOrDevice s={})
Replace NaN and infinities with finite numbers.
array flatten(const array &a, int start_axis, int end_axis=-1, StreamOrDevice s={})
Flatten the dimensions in the range [start_axis, end_axis] .
array isclose(const array &a, const array &b, double rtol=1e-5, double atol=1e-8, bool equal_nan=false, StreamOrDevice s={})
Returns a boolean array where two arrays are element-wise equal within the specified tolerance.
array operator|(const array &a, const array &b)
@@ -1387,6 +1394,7 @@ $(function() { codefold.init(0); });
array tril(array x, int k=0, StreamOrDevice s={})
array any(const array &a, bool keepdims, StreamOrDevice s={})
True if any elements in the array are true (or non-zero).
array outer(const array &a, const array &b, StreamOrDevice s={})
Compute the outer product of two vectors.
+
array hadamard_transform(const array &a, std::optional< float > scale=std::nullopt, StreamOrDevice s={})
Multiply the array by the Hadamard matrix of corresponding size.
array arcsin(const array &a, StreamOrDevice s={})
Arc Sine of the elements of an array.
array left_shift(const array &a, const array &b, StreamOrDevice s={})
Shift bits to the left.
array where(const array &condition, const array &x, const array &y, StreamOrDevice s={})
Select from x or y depending on condition.
diff --git a/docs/build/html/python/_autosummary/mlx.core.Device.html b/docs/build/html/python/_autosummary/mlx.core.Device.html
index ca9642f7b8..f01a48434d 100644
--- a/docs/build/html/python/_autosummary/mlx.core.Device.html
+++ b/docs/build/html/python/_autosummary/mlx.core.Device.html
@@ -8,7 +8,7 @@
-
mlx.core.Device — MLX 0.16.0 documentation
+
mlx.core.Device — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
-
+
@@ -131,8 +131,8 @@
-
-
+
+
@@ -296,6 +296,8 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod
+
mlx.core.einsum
+
mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv
@@ -343,6 +345,7 @@
mlx.core.minimum mlx.core.moveaxis mlx.core.multiply
+
mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones
@@ -414,6 +417,7 @@
mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform
+
mlx.core.random.laplace Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.Dtype.html b/docs/build/html/python/_autosummary/mlx.core.Dtype.html
index cf312c4837..d0dc8a5475 100644
--- a/docs/build/html/python/_autosummary/mlx.core.Dtype.html
+++ b/docs/build/html/python/_autosummary/mlx.core.Dtype.html
@@ -8,7 +8,7 @@
mlx.core.Dtype — MLX 0.16.0 documentation
+ mlx.core.Dtype — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
@@ -296,6 +296,8 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod
+
mlx.core.einsum
+
mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv
@@ -343,6 +345,7 @@
mlx.core.minimum mlx.core.moveaxis mlx.core.multiply
+
mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones
@@ -414,6 +417,7 @@
mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform
+
mlx.core.random.laplace Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.DtypeCategory.html b/docs/build/html/python/_autosummary/mlx.core.DtypeCategory.html
index 33a12eef86..77f80e27fc 100644
--- a/docs/build/html/python/_autosummary/mlx.core.DtypeCategory.html
+++ b/docs/build/html/python/_autosummary/mlx.core.DtypeCategory.html
@@ -8,7 +8,7 @@
mlx.core.DtypeCategory — MLX 0.16.0 documentation
+ mlx.core.DtypeCategory — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
@@ -296,6 +296,8 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod
+
mlx.core.einsum
+
mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv
@@ -343,6 +345,7 @@
mlx.core.minimum mlx.core.moveaxis mlx.core.multiply
+
mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones
@@ -414,6 +417,7 @@
mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform
+
mlx.core.random.laplace Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.Stream.html b/docs/build/html/python/_autosummary/mlx.core.Stream.html
index d03ecb6cc3..c686f1133a 100644
--- a/docs/build/html/python/_autosummary/mlx.core.Stream.html
+++ b/docs/build/html/python/_autosummary/mlx.core.Stream.html
@@ -8,7 +8,7 @@
mlx.core.stream — MLX 0.16.0 documentation
+ mlx.core.stream — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
@@ -296,6 +296,8 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod
+
mlx.core.einsum
+
mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv
@@ -343,6 +345,7 @@
mlx.core.minimum mlx.core.moveaxis mlx.core.multiply
+
mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones
@@ -414,6 +417,7 @@
mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform
+
mlx.core.random.laplace Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.abs.html b/docs/build/html/python/_autosummary/mlx.core.abs.html
index baf4abf42e..56718a7668 100644
--- a/docs/build/html/python/_autosummary/mlx.core.abs.html
+++ b/docs/build/html/python/_autosummary/mlx.core.abs.html
@@ -8,7 +8,7 @@
mlx.core.abs — MLX 0.16.0 documentation
+ mlx.core.abs — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
@@ -296,6 +296,8 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod
+
mlx.core.einsum
+
mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv
@@ -343,6 +345,7 @@
mlx.core.minimum mlx.core.moveaxis mlx.core.multiply
+
mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones
@@ -414,6 +417,7 @@
mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform
+
mlx.core.random.laplace Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.add.html b/docs/build/html/python/_autosummary/mlx.core.add.html
index 6f0603b892..9e2ad809f4 100644
--- a/docs/build/html/python/_autosummary/mlx.core.add.html
+++ b/docs/build/html/python/_autosummary/mlx.core.add.html
@@ -8,7 +8,7 @@
mlx.core.add — MLX 0.16.0 documentation
+ mlx.core.add — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
@@ -296,6 +296,8 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod
+
mlx.core.einsum
+
mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv
@@ -343,6 +345,7 @@
mlx.core.minimum mlx.core.moveaxis mlx.core.multiply
+
mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones
@@ -414,6 +417,7 @@
mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform
+
mlx.core.random.laplace Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.addmm.html b/docs/build/html/python/_autosummary/mlx.core.addmm.html
index 5f2e0acf56..9ebdc17cb8 100644
--- a/docs/build/html/python/_autosummary/mlx.core.addmm.html
+++ b/docs/build/html/python/_autosummary/mlx.core.addmm.html
@@ -8,7 +8,7 @@
mlx.core.addmm — MLX 0.16.0 documentation
+ mlx.core.addmm — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
@@ -296,6 +296,8 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod
+
mlx.core.einsum
+
mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv
@@ -343,6 +345,7 @@
mlx.core.minimum mlx.core.moveaxis mlx.core.multiply
+
mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones
@@ -414,6 +417,7 @@
mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform
+
mlx.core.random.laplace Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.all.html b/docs/build/html/python/_autosummary/mlx.core.all.html
index 199b635315..252fe28de2 100644
--- a/docs/build/html/python/_autosummary/mlx.core.all.html
+++ b/docs/build/html/python/_autosummary/mlx.core.all.html
@@ -8,7 +8,7 @@
mlx.core.all — MLX 0.16.0 documentation
+ mlx.core.all — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
@@ -296,6 +296,8 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod
+
mlx.core.einsum
+
mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv
@@ -343,6 +345,7 @@
mlx.core.minimum mlx.core.moveaxis mlx.core.multiply
+
mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones
@@ -414,6 +417,7 @@
mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform
+
mlx.core.random.laplace Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.allclose.html b/docs/build/html/python/_autosummary/mlx.core.allclose.html
index 03fe336e28..26a634c2a7 100644
--- a/docs/build/html/python/_autosummary/mlx.core.allclose.html
+++ b/docs/build/html/python/_autosummary/mlx.core.allclose.html
@@ -8,7 +8,7 @@
mlx.core.allclose — MLX 0.16.0 documentation
+ mlx.core.allclose — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
@@ -296,6 +296,8 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod
+
mlx.core.einsum
+
mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv
@@ -343,6 +345,7 @@
mlx.core.minimum mlx.core.moveaxis mlx.core.multiply
+
mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones
@@ -414,6 +417,7 @@
mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform
+
mlx.core.random.laplace Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.any.html b/docs/build/html/python/_autosummary/mlx.core.any.html
index eb747a5c46..04f7386ccf 100644
--- a/docs/build/html/python/_autosummary/mlx.core.any.html
+++ b/docs/build/html/python/_autosummary/mlx.core.any.html
@@ -8,7 +8,7 @@
mlx.core.any — MLX 0.16.0 documentation
+ mlx.core.any — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
@@ -296,6 +296,8 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod
+
mlx.core.einsum
+
mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv
@@ -343,6 +345,7 @@
mlx.core.minimum mlx.core.moveaxis mlx.core.multiply
+
mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones
@@ -414,6 +417,7 @@
mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform
+
mlx.core.random.laplace Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.arange.html b/docs/build/html/python/_autosummary/mlx.core.arange.html
index 6aca6e7e59..b635342dff 100644
--- a/docs/build/html/python/_autosummary/mlx.core.arange.html
+++ b/docs/build/html/python/_autosummary/mlx.core.arange.html
@@ -8,7 +8,7 @@
mlx.core.arange — MLX 0.16.0 documentation
+ mlx.core.arange — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
@@ -296,6 +296,8 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod
+
mlx.core.einsum
+
mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv
@@ -343,6 +345,7 @@
mlx.core.minimum mlx.core.moveaxis mlx.core.multiply
+
mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones
@@ -414,6 +417,7 @@
mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform
+
mlx.core.random.laplace Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.arccos.html b/docs/build/html/python/_autosummary/mlx.core.arccos.html
index 8dae08447b..19a9978c5b 100644
--- a/docs/build/html/python/_autosummary/mlx.core.arccos.html
+++ b/docs/build/html/python/_autosummary/mlx.core.arccos.html
@@ -8,7 +8,7 @@
mlx.core.arccos — MLX 0.16.0 documentation
+ mlx.core.arccos — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
@@ -296,6 +296,8 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod
+
mlx.core.einsum
+
mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv
@@ -343,6 +345,7 @@
mlx.core.minimum mlx.core.moveaxis mlx.core.multiply
+
mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones
@@ -414,6 +417,7 @@
mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform
+
mlx.core.random.laplace Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.arccosh.html b/docs/build/html/python/_autosummary/mlx.core.arccosh.html
index bf48b1f1d4..ea3d8e6602 100644
--- a/docs/build/html/python/_autosummary/mlx.core.arccosh.html
+++ b/docs/build/html/python/_autosummary/mlx.core.arccosh.html
@@ -8,7 +8,7 @@
mlx.core.arccosh — MLX 0.16.0 documentation
+ mlx.core.arccosh — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
@@ -296,6 +296,8 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod
+
mlx.core.einsum
+
mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv
@@ -343,6 +345,7 @@
mlx.core.minimum mlx.core.moveaxis mlx.core.multiply
+
mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones
@@ -414,6 +417,7 @@
mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform
+
mlx.core.random.laplace Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.arcsin.html b/docs/build/html/python/_autosummary/mlx.core.arcsin.html
index c2cfacb3c0..a4c035edb3 100644
--- a/docs/build/html/python/_autosummary/mlx.core.arcsin.html
+++ b/docs/build/html/python/_autosummary/mlx.core.arcsin.html
@@ -8,7 +8,7 @@
mlx.core.arcsin — MLX 0.16.0 documentation
+ mlx.core.arcsin — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
@@ -296,6 +296,8 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod
+
mlx.core.einsum
+
mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv
@@ -343,6 +345,7 @@
mlx.core.minimum mlx.core.moveaxis mlx.core.multiply
+
mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones
@@ -414,6 +417,7 @@
mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform
+
mlx.core.random.laplace Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.arcsinh.html b/docs/build/html/python/_autosummary/mlx.core.arcsinh.html
index 2d492d9bf7..055311582d 100644
--- a/docs/build/html/python/_autosummary/mlx.core.arcsinh.html
+++ b/docs/build/html/python/_autosummary/mlx.core.arcsinh.html
@@ -8,7 +8,7 @@
mlx.core.arcsinh — MLX 0.16.0 documentation
+ mlx.core.arcsinh — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
@@ -296,6 +296,8 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod
+
mlx.core.einsum
+
mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv
@@ -343,6 +345,7 @@
mlx.core.minimum mlx.core.moveaxis mlx.core.multiply
+
mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones
@@ -414,6 +417,7 @@
mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform
+
mlx.core.random.laplace Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.arctan.html b/docs/build/html/python/_autosummary/mlx.core.arctan.html
index 8323536182..745e6def37 100644
--- a/docs/build/html/python/_autosummary/mlx.core.arctan.html
+++ b/docs/build/html/python/_autosummary/mlx.core.arctan.html
@@ -8,7 +8,7 @@
mlx.core.arctan — MLX 0.16.0 documentation
+ mlx.core.arctan — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
@@ -296,6 +296,8 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod
+
mlx.core.einsum
+
mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv
@@ -343,6 +345,7 @@
mlx.core.minimum mlx.core.moveaxis mlx.core.multiply
+
mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones
@@ -414,6 +417,7 @@
mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform
+
mlx.core.random.laplace Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.arctan2.html b/docs/build/html/python/_autosummary/mlx.core.arctan2.html
index 76c1b62baa..73cd68f784 100644
--- a/docs/build/html/python/_autosummary/mlx.core.arctan2.html
+++ b/docs/build/html/python/_autosummary/mlx.core.arctan2.html
@@ -8,7 +8,7 @@
mlx.core.arctan2 — MLX 0.16.0 documentation
+ mlx.core.arctan2 — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
@@ -296,6 +296,8 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod
+
mlx.core.einsum
+
mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv
@@ -343,6 +345,7 @@
mlx.core.minimum mlx.core.moveaxis mlx.core.multiply
+
mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones
@@ -414,6 +417,7 @@
mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform
+
mlx.core.random.laplace Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.arctanh.html b/docs/build/html/python/_autosummary/mlx.core.arctanh.html
index 6bc2d9cb7c..3f9a8ece18 100644
--- a/docs/build/html/python/_autosummary/mlx.core.arctanh.html
+++ b/docs/build/html/python/_autosummary/mlx.core.arctanh.html
@@ -8,7 +8,7 @@
mlx.core.arctanh — MLX 0.16.0 documentation
+ mlx.core.arctanh — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
@@ -296,6 +296,8 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod
+
mlx.core.einsum
+
mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv
@@ -343,6 +345,7 @@
mlx.core.minimum mlx.core.moveaxis mlx.core.multiply
+
mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones
@@ -414,6 +417,7 @@
mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform
+
mlx.core.random.laplace Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.argmax.html b/docs/build/html/python/_autosummary/mlx.core.argmax.html
index c6c266af7d..63c198c10f 100644
--- a/docs/build/html/python/_autosummary/mlx.core.argmax.html
+++ b/docs/build/html/python/_autosummary/mlx.core.argmax.html
@@ -8,7 +8,7 @@
mlx.core.argmax — MLX 0.16.0 documentation
+ mlx.core.argmax — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
@@ -296,6 +296,8 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod
+
mlx.core.einsum
+
mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv
@@ -343,6 +345,7 @@
mlx.core.minimum mlx.core.moveaxis mlx.core.multiply
+
mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones
@@ -414,6 +417,7 @@
mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform
+
mlx.core.random.laplace Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.argmin.html b/docs/build/html/python/_autosummary/mlx.core.argmin.html
index d561389967..5f0ae28cd5 100644
--- a/docs/build/html/python/_autosummary/mlx.core.argmin.html
+++ b/docs/build/html/python/_autosummary/mlx.core.argmin.html
@@ -8,7 +8,7 @@
mlx.core.argmin — MLX 0.16.0 documentation
+ mlx.core.argmin — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
@@ -296,6 +296,8 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod
+
mlx.core.einsum
+
mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv
@@ -343,6 +345,7 @@
mlx.core.minimum mlx.core.moveaxis mlx.core.multiply
+
mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones
@@ -414,6 +417,7 @@
mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform
+
mlx.core.random.laplace Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.argpartition.html b/docs/build/html/python/_autosummary/mlx.core.argpartition.html
index 39e1467816..ac19f10a8e 100644
--- a/docs/build/html/python/_autosummary/mlx.core.argpartition.html
+++ b/docs/build/html/python/_autosummary/mlx.core.argpartition.html
@@ -8,7 +8,7 @@
mlx.core.argpartition — MLX 0.16.0 documentation
+ mlx.core.argpartition — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
@@ -296,6 +296,8 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod
+
mlx.core.einsum
+
mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv
@@ -343,6 +345,7 @@
mlx.core.minimum mlx.core.moveaxis mlx.core.multiply
+
mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones
@@ -414,6 +417,7 @@
mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform
+
mlx.core.random.laplace Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.argsort.html b/docs/build/html/python/_autosummary/mlx.core.argsort.html
index 2d8b62b7c2..71dcecc2ac 100644
--- a/docs/build/html/python/_autosummary/mlx.core.argsort.html
+++ b/docs/build/html/python/_autosummary/mlx.core.argsort.html
@@ -8,7 +8,7 @@
mlx.core.argsort — MLX 0.16.0 documentation
+ mlx.core.argsort — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
@@ -296,6 +296,8 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod
+
mlx.core.einsum
+
mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv
@@ -343,6 +345,7 @@
mlx.core.minimum mlx.core.moveaxis mlx.core.multiply
+
mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones
@@ -414,6 +417,7 @@
mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform
+
mlx.core.random.laplace Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.array.T.html b/docs/build/html/python/_autosummary/mlx.core.array.T.html
index 3a451f8a6d..872e4509e2 100644
--- a/docs/build/html/python/_autosummary/mlx.core.array.T.html
+++ b/docs/build/html/python/_autosummary/mlx.core.array.T.html
@@ -8,7 +8,7 @@
mlx.core.array.T — MLX 0.16.0 documentation
+ mlx.core.array.T — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
@@ -296,6 +296,8 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod
+
mlx.core.einsum
+
mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv
@@ -343,6 +345,7 @@
mlx.core.minimum mlx.core.moveaxis mlx.core.multiply
+
mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones
@@ -414,6 +417,7 @@
mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform
+
mlx.core.random.laplace Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.array.abs.html b/docs/build/html/python/_autosummary/mlx.core.array.abs.html
index 5014b4ee3e..632124d026 100644
--- a/docs/build/html/python/_autosummary/mlx.core.array.abs.html
+++ b/docs/build/html/python/_autosummary/mlx.core.array.abs.html
@@ -8,7 +8,7 @@
mlx.core.array.abs — MLX 0.16.0 documentation
+ mlx.core.array.abs — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
@@ -296,6 +296,8 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod
+
mlx.core.einsum
+
mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv
@@ -343,6 +345,7 @@
mlx.core.minimum mlx.core.moveaxis mlx.core.multiply
+
mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones
@@ -414,6 +417,7 @@
mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform
+
mlx.core.random.laplace Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.array.all.html b/docs/build/html/python/_autosummary/mlx.core.array.all.html
index 07407d86c2..a30ebb2e36 100644
--- a/docs/build/html/python/_autosummary/mlx.core.array.all.html
+++ b/docs/build/html/python/_autosummary/mlx.core.array.all.html
@@ -8,7 +8,7 @@
mlx.core.array.all — MLX 0.16.0 documentation
+ mlx.core.array.all — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
@@ -296,6 +296,8 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod
+
mlx.core.einsum
+
mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv
@@ -343,6 +345,7 @@
mlx.core.minimum mlx.core.moveaxis mlx.core.multiply
+
mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones
@@ -414,6 +417,7 @@
mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform
+
mlx.core.random.laplace Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.array.any.html b/docs/build/html/python/_autosummary/mlx.core.array.any.html
index 33d0ffc2fa..dc0e76884e 100644
--- a/docs/build/html/python/_autosummary/mlx.core.array.any.html
+++ b/docs/build/html/python/_autosummary/mlx.core.array.any.html
@@ -8,7 +8,7 @@
mlx.core.array.any — MLX 0.16.0 documentation
+ mlx.core.array.any — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.array.argmax.html b/docs/build/html/python/_autosummary/mlx.core.array.argmax.html
index 5e0d9e6ebb..bcb09c603b 100644
--- a/docs/build/html/python/_autosummary/mlx.core.array.argmax.html
+++ b/docs/build/html/python/_autosummary/mlx.core.array.argmax.html
@@ -8,7 +8,7 @@
mlx.core.array.argmax — MLX 0.16.0 documentation
+ mlx.core.array.argmax — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.array.argmin.html b/docs/build/html/python/_autosummary/mlx.core.array.argmin.html
index bec4b214e4..0d508b34ba 100644
--- a/docs/build/html/python/_autosummary/mlx.core.array.argmin.html
+++ b/docs/build/html/python/_autosummary/mlx.core.array.argmin.html
@@ -8,7 +8,7 @@
mlx.core.array.argmin — MLX 0.16.0 documentation
+ mlx.core.array.argmin — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.array.astype.html b/docs/build/html/python/_autosummary/mlx.core.array.astype.html
index df12f87053..57b35b4b03 100644
--- a/docs/build/html/python/_autosummary/mlx.core.array.astype.html
+++ b/docs/build/html/python/_autosummary/mlx.core.array.astype.html
@@ -8,7 +8,7 @@
mlx.core.array.astype — MLX 0.16.0 documentation
+ mlx.core.array.astype — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.array.at.html b/docs/build/html/python/_autosummary/mlx.core.array.at.html
index 67dbdcdf31..fafad459f4 100644
--- a/docs/build/html/python/_autosummary/mlx.core.array.at.html
+++ b/docs/build/html/python/_autosummary/mlx.core.array.at.html
@@ -8,7 +8,7 @@
mlx.core.array.at — MLX 0.16.0 documentation
+ mlx.core.array.at — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.array.conj.html b/docs/build/html/python/_autosummary/mlx.core.array.conj.html
index fb30af8201..e202884e89 100644
--- a/docs/build/html/python/_autosummary/mlx.core.array.conj.html
+++ b/docs/build/html/python/_autosummary/mlx.core.array.conj.html
@@ -8,7 +8,7 @@
mlx.core.array.conj — MLX 0.16.0 documentation
+ mlx.core.array.conj — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.array.cos.html b/docs/build/html/python/_autosummary/mlx.core.array.cos.html
index 8b141d03c6..9a22449280 100644
--- a/docs/build/html/python/_autosummary/mlx.core.array.cos.html
+++ b/docs/build/html/python/_autosummary/mlx.core.array.cos.html
@@ -8,7 +8,7 @@
mlx.core.array.cos — MLX 0.16.0 documentation
+ mlx.core.array.cos — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.array.cummax.html b/docs/build/html/python/_autosummary/mlx.core.array.cummax.html
index 4c38f5aa9a..8355275cca 100644
--- a/docs/build/html/python/_autosummary/mlx.core.array.cummax.html
+++ b/docs/build/html/python/_autosummary/mlx.core.array.cummax.html
@@ -8,7 +8,7 @@
mlx.core.array.cummax — MLX 0.16.0 documentation
+ mlx.core.array.cummax — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.array.cummin.html b/docs/build/html/python/_autosummary/mlx.core.array.cummin.html
index 25e5fbd27b..aa3fb6eccb 100644
--- a/docs/build/html/python/_autosummary/mlx.core.array.cummin.html
+++ b/docs/build/html/python/_autosummary/mlx.core.array.cummin.html
@@ -8,7 +8,7 @@
mlx.core.array.cummin — MLX 0.16.0 documentation
+ mlx.core.array.cummin — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.array.cumprod.html b/docs/build/html/python/_autosummary/mlx.core.array.cumprod.html
index c1dbdacd92..2ba2ab0a10 100644
--- a/docs/build/html/python/_autosummary/mlx.core.array.cumprod.html
+++ b/docs/build/html/python/_autosummary/mlx.core.array.cumprod.html
@@ -8,7 +8,7 @@
mlx.core.array.cumprod — MLX 0.16.0 documentation
+ mlx.core.array.cumprod — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.array.cumsum.html b/docs/build/html/python/_autosummary/mlx.core.array.cumsum.html
index cc4ef99b8e..d48650f473 100644
--- a/docs/build/html/python/_autosummary/mlx.core.array.cumsum.html
+++ b/docs/build/html/python/_autosummary/mlx.core.array.cumsum.html
@@ -8,7 +8,7 @@
mlx.core.array.cumsum — MLX 0.16.0 documentation
+ mlx.core.array.cumsum — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.array.diag.html b/docs/build/html/python/_autosummary/mlx.core.array.diag.html
index 8fb72eb818..d34450194f 100644
--- a/docs/build/html/python/_autosummary/mlx.core.array.diag.html
+++ b/docs/build/html/python/_autosummary/mlx.core.array.diag.html
@@ -8,7 +8,7 @@
mlx.core.array.diag — MLX 0.16.0 documentation
+ mlx.core.array.diag — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.array.diagonal.html b/docs/build/html/python/_autosummary/mlx.core.array.diagonal.html
index 520f9e7d00..0d03f7d24f 100644
--- a/docs/build/html/python/_autosummary/mlx.core.array.diagonal.html
+++ b/docs/build/html/python/_autosummary/mlx.core.array.diagonal.html
@@ -8,7 +8,7 @@
mlx.core.array.diagonal — MLX 0.16.0 documentation
+ mlx.core.array.diagonal — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.array.dtype.html b/docs/build/html/python/_autosummary/mlx.core.array.dtype.html
index 25c34c9638..af73d2a869 100644
--- a/docs/build/html/python/_autosummary/mlx.core.array.dtype.html
+++ b/docs/build/html/python/_autosummary/mlx.core.array.dtype.html
@@ -8,7 +8,7 @@
mlx.core.array.dtype — MLX 0.16.0 documentation
+ mlx.core.array.dtype — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.array.exp.html b/docs/build/html/python/_autosummary/mlx.core.array.exp.html
index 7b86224d84..68b9e8c480 100644
--- a/docs/build/html/python/_autosummary/mlx.core.array.exp.html
+++ b/docs/build/html/python/_autosummary/mlx.core.array.exp.html
@@ -8,7 +8,7 @@
mlx.core.array.exp — MLX 0.16.0 documentation
+ mlx.core.array.exp — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.array.flatten.html b/docs/build/html/python/_autosummary/mlx.core.array.flatten.html
index f4d5d58f89..2bb84828c9 100644
--- a/docs/build/html/python/_autosummary/mlx.core.array.flatten.html
+++ b/docs/build/html/python/_autosummary/mlx.core.array.flatten.html
@@ -8,7 +8,7 @@
mlx.core.array.flatten — MLX 0.16.0 documentation
+ mlx.core.array.flatten — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.array.html b/docs/build/html/python/_autosummary/mlx.core.array.html
index bbdb631e0f..6b1402dfc4 100644
--- a/docs/build/html/python/_autosummary/mlx.core.array.html
+++ b/docs/build/html/python/_autosummary/mlx.core.array.html
@@ -8,7 +8,7 @@
mlx.core.array — MLX 0.16.0 documentation
+ mlx.core.array — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.array.item.html b/docs/build/html/python/_autosummary/mlx.core.array.item.html
index ceafb2cf81..0a0a062df8 100644
--- a/docs/build/html/python/_autosummary/mlx.core.array.item.html
+++ b/docs/build/html/python/_autosummary/mlx.core.array.item.html
@@ -8,7 +8,7 @@
mlx.core.array.item — MLX 0.16.0 documentation
+ mlx.core.array.item — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.array.itemsize.html b/docs/build/html/python/_autosummary/mlx.core.array.itemsize.html
index a631f62285..dd434a0e5f 100644
--- a/docs/build/html/python/_autosummary/mlx.core.array.itemsize.html
+++ b/docs/build/html/python/_autosummary/mlx.core.array.itemsize.html
@@ -8,7 +8,7 @@
mlx.core.array.itemsize — MLX 0.16.0 documentation
+ mlx.core.array.itemsize — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.array.log.html b/docs/build/html/python/_autosummary/mlx.core.array.log.html
index ad8867a55a..b7177e450e 100644
--- a/docs/build/html/python/_autosummary/mlx.core.array.log.html
+++ b/docs/build/html/python/_autosummary/mlx.core.array.log.html
@@ -8,7 +8,7 @@
mlx.core.array.log — MLX 0.16.0 documentation
+ mlx.core.array.log — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.array.log10.html b/docs/build/html/python/_autosummary/mlx.core.array.log10.html
index 7f6f85b234..6a592025bd 100644
--- a/docs/build/html/python/_autosummary/mlx.core.array.log10.html
+++ b/docs/build/html/python/_autosummary/mlx.core.array.log10.html
@@ -8,7 +8,7 @@
mlx.core.array.log10 — MLX 0.16.0 documentation
+ mlx.core.array.log10 — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.array.log1p.html b/docs/build/html/python/_autosummary/mlx.core.array.log1p.html
index 116013d18b..9f01802890 100644
--- a/docs/build/html/python/_autosummary/mlx.core.array.log1p.html
+++ b/docs/build/html/python/_autosummary/mlx.core.array.log1p.html
@@ -8,7 +8,7 @@
mlx.core.array.log1p — MLX 0.16.0 documentation
+ mlx.core.array.log1p — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.array.log2.html b/docs/build/html/python/_autosummary/mlx.core.array.log2.html
index 7ece2ce28c..05e25fa39f 100644
--- a/docs/build/html/python/_autosummary/mlx.core.array.log2.html
+++ b/docs/build/html/python/_autosummary/mlx.core.array.log2.html
@@ -8,7 +8,7 @@
mlx.core.array.log2 — MLX 0.16.0 documentation
+ mlx.core.array.log2 — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.array.logsumexp.html b/docs/build/html/python/_autosummary/mlx.core.array.logsumexp.html
index 46ec256de8..ee072eb713 100644
--- a/docs/build/html/python/_autosummary/mlx.core.array.logsumexp.html
+++ b/docs/build/html/python/_autosummary/mlx.core.array.logsumexp.html
@@ -8,7 +8,7 @@
mlx.core.array.logsumexp — MLX 0.16.0 documentation
+ mlx.core.array.logsumexp — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.array.max.html b/docs/build/html/python/_autosummary/mlx.core.array.max.html
index 59997b23ca..60b0d8803c 100644
--- a/docs/build/html/python/_autosummary/mlx.core.array.max.html
+++ b/docs/build/html/python/_autosummary/mlx.core.array.max.html
@@ -8,7 +8,7 @@
mlx.core.array.max — MLX 0.16.0 documentation
+ mlx.core.array.max — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.array.mean.html b/docs/build/html/python/_autosummary/mlx.core.array.mean.html
index b5a76e725c..fc9b234fb2 100644
--- a/docs/build/html/python/_autosummary/mlx.core.array.mean.html
+++ b/docs/build/html/python/_autosummary/mlx.core.array.mean.html
@@ -8,7 +8,7 @@
mlx.core.array.mean — MLX 0.16.0 documentation
+ mlx.core.array.mean — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.array.min.html b/docs/build/html/python/_autosummary/mlx.core.array.min.html
index 95793fd1a4..9dfa565d94 100644
--- a/docs/build/html/python/_autosummary/mlx.core.array.min.html
+++ b/docs/build/html/python/_autosummary/mlx.core.array.min.html
@@ -8,7 +8,7 @@
mlx.core.array.min — MLX 0.16.0 documentation
+ mlx.core.array.min — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.array.moveaxis.html b/docs/build/html/python/_autosummary/mlx.core.array.moveaxis.html
index 26e9c2004f..f9c08d4e53 100644
--- a/docs/build/html/python/_autosummary/mlx.core.array.moveaxis.html
+++ b/docs/build/html/python/_autosummary/mlx.core.array.moveaxis.html
@@ -8,7 +8,7 @@
mlx.core.array.moveaxis — MLX 0.16.0 documentation
+ mlx.core.array.moveaxis — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.array.nbytes.html b/docs/build/html/python/_autosummary/mlx.core.array.nbytes.html
index 0165d7746c..e69493bc18 100644
--- a/docs/build/html/python/_autosummary/mlx.core.array.nbytes.html
+++ b/docs/build/html/python/_autosummary/mlx.core.array.nbytes.html
@@ -8,7 +8,7 @@
mlx.core.array.nbytes — MLX 0.16.0 documentation
+ mlx.core.array.nbytes — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.array.ndim.html b/docs/build/html/python/_autosummary/mlx.core.array.ndim.html
index 6e797d83c9..5f49d97d6f 100644
--- a/docs/build/html/python/_autosummary/mlx.core.array.ndim.html
+++ b/docs/build/html/python/_autosummary/mlx.core.array.ndim.html
@@ -8,7 +8,7 @@
mlx.core.array.ndim — MLX 0.16.0 documentation
+ mlx.core.array.ndim — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.array.prod.html b/docs/build/html/python/_autosummary/mlx.core.array.prod.html
index 1fd0b0053e..d1aaa4da04 100644
--- a/docs/build/html/python/_autosummary/mlx.core.array.prod.html
+++ b/docs/build/html/python/_autosummary/mlx.core.array.prod.html
@@ -8,7 +8,7 @@
mlx.core.array.prod — MLX 0.16.0 documentation
+ mlx.core.array.prod — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.array.reciprocal.html b/docs/build/html/python/_autosummary/mlx.core.array.reciprocal.html
index 7801663340..555f42b38c 100644
--- a/docs/build/html/python/_autosummary/mlx.core.array.reciprocal.html
+++ b/docs/build/html/python/_autosummary/mlx.core.array.reciprocal.html
@@ -8,7 +8,7 @@
mlx.core.array.reciprocal — MLX 0.16.0 documentation
+ mlx.core.array.reciprocal — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.array.reshape.html b/docs/build/html/python/_autosummary/mlx.core.array.reshape.html
index 47d4e52d17..2352651533 100644
--- a/docs/build/html/python/_autosummary/mlx.core.array.reshape.html
+++ b/docs/build/html/python/_autosummary/mlx.core.array.reshape.html
@@ -8,7 +8,7 @@
mlx.core.array.reshape — MLX 0.16.0 documentation
+ mlx.core.array.reshape — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.array.round.html b/docs/build/html/python/_autosummary/mlx.core.array.round.html
index f60059a76c..bf694cda40 100644
--- a/docs/build/html/python/_autosummary/mlx.core.array.round.html
+++ b/docs/build/html/python/_autosummary/mlx.core.array.round.html
@@ -8,7 +8,7 @@
mlx.core.array.round — MLX 0.16.0 documentation
+ mlx.core.array.round — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.array.rsqrt.html b/docs/build/html/python/_autosummary/mlx.core.array.rsqrt.html
index 4cc214d676..ef54ad6b80 100644
--- a/docs/build/html/python/_autosummary/mlx.core.array.rsqrt.html
+++ b/docs/build/html/python/_autosummary/mlx.core.array.rsqrt.html
@@ -8,7 +8,7 @@
mlx.core.array.rsqrt — MLX 0.16.0 documentation
+ mlx.core.array.rsqrt — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.array.shape.html b/docs/build/html/python/_autosummary/mlx.core.array.shape.html
index 32d4049f53..a2f677a6c0 100644
--- a/docs/build/html/python/_autosummary/mlx.core.array.shape.html
+++ b/docs/build/html/python/_autosummary/mlx.core.array.shape.html
@@ -8,7 +8,7 @@
mlx.core.array.shape — MLX 0.16.0 documentation
+ mlx.core.array.shape — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.array.sin.html b/docs/build/html/python/_autosummary/mlx.core.array.sin.html
index c509d94c02..8aee0638c8 100644
--- a/docs/build/html/python/_autosummary/mlx.core.array.sin.html
+++ b/docs/build/html/python/_autosummary/mlx.core.array.sin.html
@@ -8,7 +8,7 @@
mlx.core.array.sin — MLX 0.16.0 documentation
+ mlx.core.array.sin — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.array.size.html b/docs/build/html/python/_autosummary/mlx.core.array.size.html
index cc74417c04..5f447c7e54 100644
--- a/docs/build/html/python/_autosummary/mlx.core.array.size.html
+++ b/docs/build/html/python/_autosummary/mlx.core.array.size.html
@@ -8,7 +8,7 @@
mlx.core.array.size — MLX 0.16.0 documentation
+ mlx.core.array.size — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.array.split.html b/docs/build/html/python/_autosummary/mlx.core.array.split.html
index f06c9a471b..d9564129a8 100644
--- a/docs/build/html/python/_autosummary/mlx.core.array.split.html
+++ b/docs/build/html/python/_autosummary/mlx.core.array.split.html
@@ -8,7 +8,7 @@
mlx.core.array.split — MLX 0.16.0 documentation
+ mlx.core.array.split — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.array.sqrt.html b/docs/build/html/python/_autosummary/mlx.core.array.sqrt.html
index 9457bcc8a3..b702172f78 100644
--- a/docs/build/html/python/_autosummary/mlx.core.array.sqrt.html
+++ b/docs/build/html/python/_autosummary/mlx.core.array.sqrt.html
@@ -8,7 +8,7 @@
mlx.core.array.sqrt — MLX 0.16.0 documentation
+ mlx.core.array.sqrt — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.array.square.html b/docs/build/html/python/_autosummary/mlx.core.array.square.html
index 091c2eabf2..b087ee92e4 100644
--- a/docs/build/html/python/_autosummary/mlx.core.array.square.html
+++ b/docs/build/html/python/_autosummary/mlx.core.array.square.html
@@ -8,7 +8,7 @@
mlx.core.array.square — MLX 0.16.0 documentation
+ mlx.core.array.square — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.array.squeeze.html b/docs/build/html/python/_autosummary/mlx.core.array.squeeze.html
index 91b6c4ccee..7ce553bf90 100644
--- a/docs/build/html/python/_autosummary/mlx.core.array.squeeze.html
+++ b/docs/build/html/python/_autosummary/mlx.core.array.squeeze.html
@@ -8,7 +8,7 @@
mlx.core.array.squeeze — MLX 0.16.0 documentation
+ mlx.core.array.squeeze — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.array.sum.html b/docs/build/html/python/_autosummary/mlx.core.array.sum.html
index 584cae5b63..ba947b71ef 100644
--- a/docs/build/html/python/_autosummary/mlx.core.array.sum.html
+++ b/docs/build/html/python/_autosummary/mlx.core.array.sum.html
@@ -8,7 +8,7 @@
mlx.core.array.sum — MLX 0.16.0 documentation
+ mlx.core.array.sum — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.array.swapaxes.html b/docs/build/html/python/_autosummary/mlx.core.array.swapaxes.html
index ecb1083f09..9bd63f45cc 100644
--- a/docs/build/html/python/_autosummary/mlx.core.array.swapaxes.html
+++ b/docs/build/html/python/_autosummary/mlx.core.array.swapaxes.html
@@ -8,7 +8,7 @@
mlx.core.array.swapaxes — MLX 0.16.0 documentation
+ mlx.core.array.swapaxes — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.array.tolist.html b/docs/build/html/python/_autosummary/mlx.core.array.tolist.html
index 25394470bc..d9eb275d82 100644
--- a/docs/build/html/python/_autosummary/mlx.core.array.tolist.html
+++ b/docs/build/html/python/_autosummary/mlx.core.array.tolist.html
@@ -8,7 +8,7 @@
mlx.core.array.tolist — MLX 0.16.0 documentation
+ mlx.core.array.tolist — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.array.transpose.html b/docs/build/html/python/_autosummary/mlx.core.array.transpose.html
index ea90e22bf9..5c825776d4 100644
--- a/docs/build/html/python/_autosummary/mlx.core.array.transpose.html
+++ b/docs/build/html/python/_autosummary/mlx.core.array.transpose.html
@@ -8,7 +8,7 @@
mlx.core.array.transpose — MLX 0.16.0 documentation
+ mlx.core.array.transpose — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.array.var.html b/docs/build/html/python/_autosummary/mlx.core.array.var.html
index 1c7ecd0eaf..fbcfb302e0 100644
--- a/docs/build/html/python/_autosummary/mlx.core.array.var.html
+++ b/docs/build/html/python/_autosummary/mlx.core.array.var.html
@@ -8,7 +8,7 @@
mlx.core.array.var — MLX 0.16.0 documentation
+ mlx.core.array.var — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.array.view.html b/docs/build/html/python/_autosummary/mlx.core.array.view.html
index c44109836a..d01f75ef6a 100644
--- a/docs/build/html/python/_autosummary/mlx.core.array.view.html
+++ b/docs/build/html/python/_autosummary/mlx.core.array.view.html
@@ -8,7 +8,7 @@
mlx.core.array.view — MLX 0.16.0 documentation
+ mlx.core.array.view — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.array_equal.html b/docs/build/html/python/_autosummary/mlx.core.array_equal.html
index 56877909e9..e230633f3c 100644
--- a/docs/build/html/python/_autosummary/mlx.core.array_equal.html
+++ b/docs/build/html/python/_autosummary/mlx.core.array_equal.html
@@ -8,7 +8,7 @@
mlx.core.array_equal — MLX 0.16.0 documentation
+ mlx.core.array_equal — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.as_strided.html b/docs/build/html/python/_autosummary/mlx.core.as_strided.html
index 22788236c4..7a95e96f8f 100644
--- a/docs/build/html/python/_autosummary/mlx.core.as_strided.html
+++ b/docs/build/html/python/_autosummary/mlx.core.as_strided.html
@@ -8,7 +8,7 @@
mlx.core.as_strided — MLX 0.16.0 documentation
+ mlx.core.as_strided — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.atleast_1d.html b/docs/build/html/python/_autosummary/mlx.core.atleast_1d.html
index 8afeaf2624..49f35c0ae9 100644
--- a/docs/build/html/python/_autosummary/mlx.core.atleast_1d.html
+++ b/docs/build/html/python/_autosummary/mlx.core.atleast_1d.html
@@ -8,7 +8,7 @@
mlx.core.atleast_1d — MLX 0.16.0 documentation
+ mlx.core.atleast_1d — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.atleast_2d.html b/docs/build/html/python/_autosummary/mlx.core.atleast_2d.html
index 065f85990c..aec628a4c6 100644
--- a/docs/build/html/python/_autosummary/mlx.core.atleast_2d.html
+++ b/docs/build/html/python/_autosummary/mlx.core.atleast_2d.html
@@ -8,7 +8,7 @@
mlx.core.atleast_2d — MLX 0.16.0 documentation
+ mlx.core.atleast_2d — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.atleast_3d.html b/docs/build/html/python/_autosummary/mlx.core.atleast_3d.html
index c93b1970c1..4e0c9a9c8e 100644
--- a/docs/build/html/python/_autosummary/mlx.core.atleast_3d.html
+++ b/docs/build/html/python/_autosummary/mlx.core.atleast_3d.html
@@ -8,7 +8,7 @@
mlx.core.atleast_3d — MLX 0.16.0 documentation
+ mlx.core.atleast_3d — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.bitwise_and.html b/docs/build/html/python/_autosummary/mlx.core.bitwise_and.html
index 5c5b0a2358..232760d85a 100644
--- a/docs/build/html/python/_autosummary/mlx.core.bitwise_and.html
+++ b/docs/build/html/python/_autosummary/mlx.core.bitwise_and.html
@@ -8,7 +8,7 @@
mlx.core.bitwise_and — MLX 0.16.0 documentation
+ mlx.core.bitwise_and — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.bitwise_or.html b/docs/build/html/python/_autosummary/mlx.core.bitwise_or.html
index e15a58ad2c..7cb7ccc43a 100644
--- a/docs/build/html/python/_autosummary/mlx.core.bitwise_or.html
+++ b/docs/build/html/python/_autosummary/mlx.core.bitwise_or.html
@@ -8,7 +8,7 @@
mlx.core.bitwise_or — MLX 0.16.0 documentation
+ mlx.core.bitwise_or — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.bitwise_xor.html b/docs/build/html/python/_autosummary/mlx.core.bitwise_xor.html
index 6f0c782e4a..6c1e7f840b 100644
--- a/docs/build/html/python/_autosummary/mlx.core.bitwise_xor.html
+++ b/docs/build/html/python/_autosummary/mlx.core.bitwise_xor.html
@@ -8,7 +8,7 @@
mlx.core.bitwise_xor — MLX 0.16.0 documentation
+ mlx.core.bitwise_xor — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.block_masked_mm.html b/docs/build/html/python/_autosummary/mlx.core.block_masked_mm.html
index 69ca026c72..38f5fc8c69 100644
--- a/docs/build/html/python/_autosummary/mlx.core.block_masked_mm.html
+++ b/docs/build/html/python/_autosummary/mlx.core.block_masked_mm.html
@@ -8,7 +8,7 @@
mlx.core.block_masked_mm — MLX 0.16.0 documentation
+ mlx.core.block_masked_mm — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
@@ -859,12 +863,18 @@ of size block_size<
a (array
b (array
block_size (int 32 64 64
mask_out (array , optional ) – Mask for output (default: None
mask_lhs (array , optional ) – Mask for a (default: None
mask_rhs (array , optional ) – Mask for b (default: None
block_size (int 32 64 64
mask_out (array , optional ) – Mask for output. Default: None
mask_lhs (array , optional ) – Mask for a None
mask_rhs (array , optional ) – Mask for b None
Returns:
+The output array.
+Return type:
+array
+mlx.core.broadcast_to — MLX 0.16.0 documentation
+ mlx.core.broadcast_to — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.ceil.html b/docs/build/html/python/_autosummary/mlx.core.ceil.html
index efe3561ed8..303647e190 100644
--- a/docs/build/html/python/_autosummary/mlx.core.ceil.html
+++ b/docs/build/html/python/_autosummary/mlx.core.ceil.html
@@ -8,7 +8,7 @@
mlx.core.ceil — MLX 0.16.0 documentation
+ mlx.core.ceil — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.clip.html b/docs/build/html/python/_autosummary/mlx.core.clip.html
index f203ddc303..866979a34b 100644
--- a/docs/build/html/python/_autosummary/mlx.core.clip.html
+++ b/docs/build/html/python/_autosummary/mlx.core.clip.html
@@ -8,7 +8,7 @@
mlx.core.clip — MLX 0.16.0 documentation
+ mlx.core.clip — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.compile.html b/docs/build/html/python/_autosummary/mlx.core.compile.html
index 5a568fa724..bca6bb2c2e 100644
--- a/docs/build/html/python/_autosummary/mlx.core.compile.html
+++ b/docs/build/html/python/_autosummary/mlx.core.compile.html
@@ -8,7 +8,7 @@
mlx.core.compile — MLX 0.16.0 documentation
+ mlx.core.compile — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.concatenate.html b/docs/build/html/python/_autosummary/mlx.core.concatenate.html
index 5ed28a1ca1..88f4b6c377 100644
--- a/docs/build/html/python/_autosummary/mlx.core.concatenate.html
+++ b/docs/build/html/python/_autosummary/mlx.core.concatenate.html
@@ -8,7 +8,7 @@
mlx.core.concatenate — MLX 0.16.0 documentation
+ mlx.core.concatenate — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.conj.html b/docs/build/html/python/_autosummary/mlx.core.conj.html
index 974159da7f..26ebae2f8a 100644
--- a/docs/build/html/python/_autosummary/mlx.core.conj.html
+++ b/docs/build/html/python/_autosummary/mlx.core.conj.html
@@ -8,7 +8,7 @@
mlx.core.conj — MLX 0.16.0 documentation
+ mlx.core.conj — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
@@ -849,6 +853,12 @@ Alias for mx.conjugate .
Parameters:
a (array
Returns:
+The output array.
+Return type:
+array
+mlx.core.conjugate — MLX 0.16.0 documentation
+ mlx.core.conjugate — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
@@ -849,6 +853,12 @@ Alias for mx.conj .
Parameters:
a (array
Returns:
+The output array.
+Return type:
+array
+mlx.core.conv1d — MLX 0.16.0 documentation
+ mlx.core.conv1d — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.conv2d.html b/docs/build/html/python/_autosummary/mlx.core.conv2d.html
index 2c128e1a42..f4bf1351d1 100644
--- a/docs/build/html/python/_autosummary/mlx.core.conv2d.html
+++ b/docs/build/html/python/_autosummary/mlx.core.conv2d.html
@@ -8,7 +8,7 @@
mlx.core.conv2d — MLX 0.16.0 documentation
+ mlx.core.conv2d — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.conv_general.html b/docs/build/html/python/_autosummary/mlx.core.conv_general.html
index 8550aafa72..629625594c 100644
--- a/docs/build/html/python/_autosummary/mlx.core.conv_general.html
+++ b/docs/build/html/python/_autosummary/mlx.core.conv_general.html
@@ -8,7 +8,7 @@
mlx.core.conv_general — MLX 0.16.0 documentation
+ mlx.core.conv_general — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.convolve.html b/docs/build/html/python/_autosummary/mlx.core.convolve.html
index c06429b02d..19a1870e0e 100644
--- a/docs/build/html/python/_autosummary/mlx.core.convolve.html
+++ b/docs/build/html/python/_autosummary/mlx.core.convolve.html
@@ -8,7 +8,7 @@
mlx.core.convolve — MLX 0.16.0 documentation
+ mlx.core.convolve — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.cos.html b/docs/build/html/python/_autosummary/mlx.core.cos.html
index 51063843d1..fdd1b84581 100644
--- a/docs/build/html/python/_autosummary/mlx.core.cos.html
+++ b/docs/build/html/python/_autosummary/mlx.core.cos.html
@@ -8,7 +8,7 @@
mlx.core.cos — MLX 0.16.0 documentation
+ mlx.core.cos — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.cosh.html b/docs/build/html/python/_autosummary/mlx.core.cosh.html
index 43b1a30f2f..2ba021b08d 100644
--- a/docs/build/html/python/_autosummary/mlx.core.cosh.html
+++ b/docs/build/html/python/_autosummary/mlx.core.cosh.html
@@ -8,7 +8,7 @@
mlx.core.cosh — MLX 0.16.0 documentation
+ mlx.core.cosh — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.cummax.html b/docs/build/html/python/_autosummary/mlx.core.cummax.html
index 68bed456fe..ebcb72beb2 100644
--- a/docs/build/html/python/_autosummary/mlx.core.cummax.html
+++ b/docs/build/html/python/_autosummary/mlx.core.cummax.html
@@ -8,7 +8,7 @@
mlx.core.cummax — MLX 0.16.0 documentation
+ mlx.core.cummax — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
@@ -856,6 +860,12 @@ returned. Returns:
+The output array.
+Return type:
+array
+mlx.core.cummin — MLX 0.16.0 documentation
+ mlx.core.cummin — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace Transforms
@@ -856,6 +860,12 @@ returned. Returns:
+The output array.
+Return type:
+array
+mlx.core.cumprod — MLX 0.16.0 documentation
+ mlx.core.cumprod — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace Transforms
@@ -856,6 +860,12 @@ returned. Returns:
+The output array.
+Return type:
+array
+mlx.core.cumsum — MLX 0.16.0 documentation
+ mlx.core.cumsum — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace Transforms
@@ -856,6 +860,12 @@ returned. Returns:
+The output array.
+Return type:
+array
+mlx.core.custom_function — MLX 0.16.0 documentation
+ mlx.core.custom_function — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace Transforms
@@ -851,7 +855,7 @@ document.write(`
callables that behave identically to the wrapped function. However, when
a function transformation is used (e.g. computing gradients using
value_and_grad() custom_function.vjp() custom_function.jvp() custom_function.vjp() custom_function.jvp() custom_function.vmap() Note, all custom transformations are optional. Undefined transformations
fall back to the default behaviour.
@@ -893,7 +897,7 @@ fall back to the default behaviour.
__init__
jvp
jvp
Define a custom jvp for the wrapped function.
vjp
mlx.core.default_device — MLX 0.16.0 documentation
+ mlx.core.default_device — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.default_stream.html b/docs/build/html/python/_autosummary/mlx.core.default_stream.html
index 472747e523..3785f39609 100644
--- a/docs/build/html/python/_autosummary/mlx.core.default_stream.html
+++ b/docs/build/html/python/_autosummary/mlx.core.default_stream.html
@@ -8,7 +8,7 @@
mlx.core.default_stream — MLX 0.16.0 documentation
+ mlx.core.default_stream — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.degrees.html b/docs/build/html/python/_autosummary/mlx.core.degrees.html
index f21f89efd0..7e43552d09 100644
--- a/docs/build/html/python/_autosummary/mlx.core.degrees.html
+++ b/docs/build/html/python/_autosummary/mlx.core.degrees.html
@@ -8,7 +8,7 @@
mlx.core.degrees — MLX 0.16.0 documentation
+ mlx.core.degrees — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.dequantize.html b/docs/build/html/python/_autosummary/mlx.core.dequantize.html
index 6bf121e3c7..7b90b59e1c 100644
--- a/docs/build/html/python/_autosummary/mlx.core.dequantize.html
+++ b/docs/build/html/python/_autosummary/mlx.core.dequantize.html
@@ -8,7 +8,7 @@
mlx.core.dequantize — MLX 0.16.0 documentation
+ mlx.core.dequantize — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
@@ -858,9 +862,9 @@ document.write(`
scales (array group_size w
biases (array group_size w
group_size (int , optional ) – The size of the group in w 64
64 bits (int , optional ) – The number of bits occupied by each element in
-w 4
w 4 Returns:
diff --git a/docs/build/html/python/_autosummary/mlx.core.diag.html b/docs/build/html/python/_autosummary/mlx.core.diag.html
index 4a2ccbf28c..38d5c5cef8 100644
--- a/docs/build/html/python/_autosummary/mlx.core.diag.html
+++ b/docs/build/html/python/_autosummary/mlx.core.diag.html
@@ -8,7 +8,7 @@
mlx.core.diag — MLX 0.16.0 documentation
+ mlx.core.diag — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.diagonal.html b/docs/build/html/python/_autosummary/mlx.core.diagonal.html
index 3cbd503b5e..ee0b125d6d 100644
--- a/docs/build/html/python/_autosummary/mlx.core.diagonal.html
+++ b/docs/build/html/python/_autosummary/mlx.core.diagonal.html
@@ -8,7 +8,7 @@
mlx.core.diagonal — MLX 0.16.0 documentation
+ mlx.core.diagonal — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.disable_compile.html b/docs/build/html/python/_autosummary/mlx.core.disable_compile.html
index c35526ea42..724855bc01 100644
--- a/docs/build/html/python/_autosummary/mlx.core.disable_compile.html
+++ b/docs/build/html/python/_autosummary/mlx.core.disable_compile.html
@@ -8,7 +8,7 @@
mlx.core.disable_compile — MLX 0.16.0 documentation
+ mlx.core.disable_compile — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.distributed.Group.html b/docs/build/html/python/_autosummary/mlx.core.distributed.Group.html
index 33e2f80820..dfc8231fe5 100644
--- a/docs/build/html/python/_autosummary/mlx.core.distributed.Group.html
+++ b/docs/build/html/python/_autosummary/mlx.core.distributed.Group.html
@@ -8,7 +8,7 @@
mlx.core.distributed.Group — MLX 0.16.0 documentation
+ mlx.core.distributed.Group — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.distributed.all_gather.html b/docs/build/html/python/_autosummary/mlx.core.distributed.all_gather.html
index 629972226e..c4b0a295d6 100644
--- a/docs/build/html/python/_autosummary/mlx.core.distributed.all_gather.html
+++ b/docs/build/html/python/_autosummary/mlx.core.distributed.all_gather.html
@@ -8,7 +8,7 @@
mlx.core.distributed.all_gather — MLX 0.16.0 documentation
+ mlx.core.distributed.all_gather — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.distributed.all_sum.html b/docs/build/html/python/_autosummary/mlx.core.distributed.all_sum.html
index 991b23c989..7d7e525735 100644
--- a/docs/build/html/python/_autosummary/mlx.core.distributed.all_sum.html
+++ b/docs/build/html/python/_autosummary/mlx.core.distributed.all_sum.html
@@ -8,7 +8,7 @@
mlx.core.distributed.all_sum — MLX 0.16.0 documentation
+ mlx.core.distributed.all_sum — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.distributed.init.html b/docs/build/html/python/_autosummary/mlx.core.distributed.init.html
index 88fdddbc4c..71b0b064a0 100644
--- a/docs/build/html/python/_autosummary/mlx.core.distributed.init.html
+++ b/docs/build/html/python/_autosummary/mlx.core.distributed.init.html
@@ -8,7 +8,7 @@
mlx.core.distributed.init — MLX 0.16.0 documentation
+ mlx.core.distributed.init — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.distributed.is_available.html b/docs/build/html/python/_autosummary/mlx.core.distributed.is_available.html
index fb98de44b1..b1f637a72d 100644
--- a/docs/build/html/python/_autosummary/mlx.core.distributed.is_available.html
+++ b/docs/build/html/python/_autosummary/mlx.core.distributed.is_available.html
@@ -8,7 +8,7 @@
mlx.core.distributed.is_available — MLX 0.16.0 documentation
+ mlx.core.distributed.is_available — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.divide.html b/docs/build/html/python/_autosummary/mlx.core.divide.html
index 60a41cd51d..3c07cbe644 100644
--- a/docs/build/html/python/_autosummary/mlx.core.divide.html
+++ b/docs/build/html/python/_autosummary/mlx.core.divide.html
@@ -8,7 +8,7 @@
mlx.core.divide — MLX 0.16.0 documentation
+ mlx.core.divide — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.divmod.html b/docs/build/html/python/_autosummary/mlx.core.divmod.html
index 48167d2d5d..fc4f3a1af6 100644
--- a/docs/build/html/python/_autosummary/mlx.core.divmod.html
+++ b/docs/build/html/python/_autosummary/mlx.core.divmod.html
@@ -8,7 +8,7 @@
mlx.core.divmod — MLX 0.16.0 documentation
+ mlx.core.divmod — MLX 0.16.1 documentation
@@ -36,14 +36,14 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
@@ -886,11 +890,11 @@ semantics. Either or both input arrays can also be scalars.
next
-
mlx.core.equal
+
mlx.core.einsum
diff --git a/docs/build/html/python/_autosummary/mlx.core.custom_function.jvp.html b/docs/build/html/python/_autosummary/mlx.core.einsum.html
similarity index 93%
rename from docs/build/html/python/_autosummary/mlx.core.custom_function.jvp.html
rename to docs/build/html/python/_autosummary/mlx.core.einsum.html
index 6e3d42e691..f52dfa57cb 100644
--- a/docs/build/html/python/_autosummary/mlx.core.custom_function.jvp.html
+++ b/docs/build/html/python/_autosummary/mlx.core.einsum.html
@@ -8,7 +8,7 @@
mlx.core.custom_function.jvp — MLX 0.16.0 documentation
+ mlx.core.einsum — MLX 0.16.1 documentation
@@ -36,13 +36,15 @@
LLM inference
Python API Reference
-Returns:
diff --git a/docs/build/html/python/_autosummary/mlx.core.grad.html b/docs/build/html/python/_autosummary/mlx.core.grad.html
index dbedc8371e..3280cce3ff 100644
--- a/docs/build/html/python/_autosummary/mlx.core.grad.html
+++ b/docs/build/html/python/_autosummary/mlx.core.grad.html
@@ -8,7 +8,7 @@
mlx.core.grad — MLX 0.16.0 documentation
+ mlx.core.grad — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.greater.html b/docs/build/html/python/_autosummary/mlx.core.greater.html
index 8118ce6d2c..839d448e21 100644
--- a/docs/build/html/python/_autosummary/mlx.core.greater.html
+++ b/docs/build/html/python/_autosummary/mlx.core.greater.html
@@ -8,7 +8,7 @@
mlx.core.greater — MLX 0.16.0 documentation
+ mlx.core.greater — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.greater_equal.html b/docs/build/html/python/_autosummary/mlx.core.greater_equal.html
index bdd32afa01..a7414d144c 100644
--- a/docs/build/html/python/_autosummary/mlx.core.greater_equal.html
+++ b/docs/build/html/python/_autosummary/mlx.core.greater_equal.html
@@ -8,7 +8,7 @@
mlx.core.greater_equal — MLX 0.16.0 documentation
+ mlx.core.greater_equal — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.hadamard_transform.html b/docs/build/html/python/_autosummary/mlx.core.hadamard_transform.html
index 7482de7e0c..bc2e802807 100644
--- a/docs/build/html/python/_autosummary/mlx.core.hadamard_transform.html
+++ b/docs/build/html/python/_autosummary/mlx.core.hadamard_transform.html
@@ -8,7 +8,7 @@
mlx.core.hadamard_transform — MLX 0.16.0 documentation
+ mlx.core.hadamard_transform — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
@@ -842,12 +846,12 @@ document.write(`
mlx.core.hadamard_transform
-hadamard_transform ( a: array float scale = 1.0 stream: Union[None Stream Device] = None ) → array
+hadamard_transform ( a: array Optional[float] scale = None stream: Union[None Stream Device] = None ) → array Perform the Walsh-Hadamard transform along the final axis.
Equivalent to:
from scipy.linalg import hadamard
-y = hadamard ( len ( x )) @ x
+y = ( hadamard ( len ( x )) @ x ) * scale
Supports sizes n = m*2^k m (1, 12, 20, 28) 2^k
@@ -856,7 +860,8 @@ document.write(`
Parameters:
a (array
scale (float
scale (float 1/sqrt(a.shape[-1])
Returns:
diff --git a/docs/build/html/python/_autosummary/mlx.core.identity.html b/docs/build/html/python/_autosummary/mlx.core.identity.html
index cc5b876053..c387f8b7ed 100644
--- a/docs/build/html/python/_autosummary/mlx.core.identity.html
+++ b/docs/build/html/python/_autosummary/mlx.core.identity.html
@@ -8,7 +8,7 @@
mlx.core.identity — MLX 0.16.0 documentation
+ mlx.core.identity — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.inner.html b/docs/build/html/python/_autosummary/mlx.core.inner.html
index b3dd36be25..f13e4d11f5 100644
--- a/docs/build/html/python/_autosummary/mlx.core.inner.html
+++ b/docs/build/html/python/_autosummary/mlx.core.inner.html
@@ -8,7 +8,7 @@
mlx.core.inner — MLX 0.16.0 documentation
+ mlx.core.inner — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.isclose.html b/docs/build/html/python/_autosummary/mlx.core.isclose.html
index 0b19167ee7..abb0566bc0 100644
--- a/docs/build/html/python/_autosummary/mlx.core.isclose.html
+++ b/docs/build/html/python/_autosummary/mlx.core.isclose.html
@@ -8,7 +8,7 @@
mlx.core.isclose — MLX 0.16.0 documentation
+ mlx.core.isclose — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.isinf.html b/docs/build/html/python/_autosummary/mlx.core.isinf.html
index eed06eaf94..f465f25bc2 100644
--- a/docs/build/html/python/_autosummary/mlx.core.isinf.html
+++ b/docs/build/html/python/_autosummary/mlx.core.isinf.html
@@ -8,7 +8,7 @@
mlx.core.isinf — MLX 0.16.0 documentation
+ mlx.core.isinf — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.isnan.html b/docs/build/html/python/_autosummary/mlx.core.isnan.html
index 9a95b47513..997249e91c 100644
--- a/docs/build/html/python/_autosummary/mlx.core.isnan.html
+++ b/docs/build/html/python/_autosummary/mlx.core.isnan.html
@@ -8,7 +8,7 @@
mlx.core.isnan — MLX 0.16.0 documentation
+ mlx.core.isnan — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.isneginf.html b/docs/build/html/python/_autosummary/mlx.core.isneginf.html
index 3fe6679ad0..c64637e42d 100644
--- a/docs/build/html/python/_autosummary/mlx.core.isneginf.html
+++ b/docs/build/html/python/_autosummary/mlx.core.isneginf.html
@@ -8,7 +8,7 @@
mlx.core.isneginf — MLX 0.16.0 documentation
+ mlx.core.isneginf — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.isposinf.html b/docs/build/html/python/_autosummary/mlx.core.isposinf.html
index a6be6bd51d..0944c9da5e 100644
--- a/docs/build/html/python/_autosummary/mlx.core.isposinf.html
+++ b/docs/build/html/python/_autosummary/mlx.core.isposinf.html
@@ -8,7 +8,7 @@
mlx.core.isposinf — MLX 0.16.0 documentation
+ mlx.core.isposinf — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.issubdtype.html b/docs/build/html/python/_autosummary/mlx.core.issubdtype.html
index a9fb59b7c1..bc506704a4 100644
--- a/docs/build/html/python/_autosummary/mlx.core.issubdtype.html
+++ b/docs/build/html/python/_autosummary/mlx.core.issubdtype.html
@@ -8,7 +8,7 @@
mlx.core.issubdtype — MLX 0.16.0 documentation
+ mlx.core.issubdtype — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.jvp.html b/docs/build/html/python/_autosummary/mlx.core.jvp.html
index 7ae680e390..70f90d7f40 100644
--- a/docs/build/html/python/_autosummary/mlx.core.jvp.html
+++ b/docs/build/html/python/_autosummary/mlx.core.jvp.html
@@ -8,7 +8,7 @@
mlx.core.jvp — MLX 0.16.0 documentation
+ mlx.core.jvp — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.left_shift.html b/docs/build/html/python/_autosummary/mlx.core.left_shift.html
index 01b576c111..7011b396c5 100644
--- a/docs/build/html/python/_autosummary/mlx.core.left_shift.html
+++ b/docs/build/html/python/_autosummary/mlx.core.left_shift.html
@@ -8,7 +8,7 @@
mlx.core.left_shift — MLX 0.16.0 documentation
+ mlx.core.left_shift — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.less.html b/docs/build/html/python/_autosummary/mlx.core.less.html
index f59268a164..41310b195b 100644
--- a/docs/build/html/python/_autosummary/mlx.core.less.html
+++ b/docs/build/html/python/_autosummary/mlx.core.less.html
@@ -8,7 +8,7 @@
mlx.core.less — MLX 0.16.0 documentation
+ mlx.core.less — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.less_equal.html b/docs/build/html/python/_autosummary/mlx.core.less_equal.html
index 71139e207d..fc300c15de 100644
--- a/docs/build/html/python/_autosummary/mlx.core.less_equal.html
+++ b/docs/build/html/python/_autosummary/mlx.core.less_equal.html
@@ -8,7 +8,7 @@
mlx.core.less_equal — MLX 0.16.0 documentation
+ mlx.core.less_equal — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.linalg.cholesky.html b/docs/build/html/python/_autosummary/mlx.core.linalg.cholesky.html
index 83b6278a8a..cf49cfa376 100644
--- a/docs/build/html/python/_autosummary/mlx.core.linalg.cholesky.html
+++ b/docs/build/html/python/_autosummary/mlx.core.linalg.cholesky.html
@@ -8,7 +8,7 @@
mlx.core.linalg.cholesky — MLX 0.16.0 documentation
+ mlx.core.linalg.cholesky — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.linalg.inv.html b/docs/build/html/python/_autosummary/mlx.core.linalg.inv.html
index 2064603c85..29a300697d 100644
--- a/docs/build/html/python/_autosummary/mlx.core.linalg.inv.html
+++ b/docs/build/html/python/_autosummary/mlx.core.linalg.inv.html
@@ -8,7 +8,7 @@
mlx.core.linalg.inv — MLX 0.16.0 documentation
+ mlx.core.linalg.inv — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.linalg.norm.html b/docs/build/html/python/_autosummary/mlx.core.linalg.norm.html
index 3eddcadbe6..b41acbd92e 100644
--- a/docs/build/html/python/_autosummary/mlx.core.linalg.norm.html
+++ b/docs/build/html/python/_autosummary/mlx.core.linalg.norm.html
@@ -8,7 +8,7 @@
mlx.core.linalg.norm — MLX 0.16.0 documentation
+ mlx.core.linalg.norm — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.linalg.qr.html b/docs/build/html/python/_autosummary/mlx.core.linalg.qr.html
index aae55b83c6..2b09a2a935 100644
--- a/docs/build/html/python/_autosummary/mlx.core.linalg.qr.html
+++ b/docs/build/html/python/_autosummary/mlx.core.linalg.qr.html
@@ -8,7 +8,7 @@
mlx.core.linalg.qr — MLX 0.16.0 documentation
+ mlx.core.linalg.qr — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.linalg.svd.html b/docs/build/html/python/_autosummary/mlx.core.linalg.svd.html
index fb33f5b579..ac5e8586e3 100644
--- a/docs/build/html/python/_autosummary/mlx.core.linalg.svd.html
+++ b/docs/build/html/python/_autosummary/mlx.core.linalg.svd.html
@@ -8,7 +8,7 @@
mlx.core.linalg.svd — MLX 0.16.0 documentation
+ mlx.core.linalg.svd — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.linspace.html b/docs/build/html/python/_autosummary/mlx.core.linspace.html
index abc617ba00..b17ade6342 100644
--- a/docs/build/html/python/_autosummary/mlx.core.linspace.html
+++ b/docs/build/html/python/_autosummary/mlx.core.linspace.html
@@ -8,7 +8,7 @@
mlx.core.linspace — MLX 0.16.0 documentation
+ mlx.core.linspace — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.load.html b/docs/build/html/python/_autosummary/mlx.core.load.html
index 7986e2ae83..0abeaaaacf 100644
--- a/docs/build/html/python/_autosummary/mlx.core.load.html
+++ b/docs/build/html/python/_autosummary/mlx.core.load.html
@@ -8,7 +8,7 @@
mlx.core.load — MLX 0.16.0 documentation
+ mlx.core.load — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
@@ -850,13 +854,12 @@ document.write(`
Parameters:
file (file , str
format (str , optional ) – Format of the file. If None
format – is inferred from the file extension. Supported formats:
-npy npz safetensors None
return_metadata (bool , optional ) – Load the metadata for formats
which – support matadata. The metadata will be returned as an
-additional dictionary.
format (str , optional ) – Format of the file. If None npy npz safetensors None
return_metadata (bool , optional ) – Load the metadata for formats
+which support matadata. The metadata will be returned as an
+additional dictionary. Default: False
Returns:
diff --git a/docs/build/html/python/_autosummary/mlx.core.log.html b/docs/build/html/python/_autosummary/mlx.core.log.html
index bf82976a4b..2ccfa05db7 100644
--- a/docs/build/html/python/_autosummary/mlx.core.log.html
+++ b/docs/build/html/python/_autosummary/mlx.core.log.html
@@ -8,7 +8,7 @@
mlx.core.log — MLX 0.16.0 documentation
+ mlx.core.log — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.log10.html b/docs/build/html/python/_autosummary/mlx.core.log10.html
index f903e58130..99f4242852 100644
--- a/docs/build/html/python/_autosummary/mlx.core.log10.html
+++ b/docs/build/html/python/_autosummary/mlx.core.log10.html
@@ -8,7 +8,7 @@
mlx.core.log10 — MLX 0.16.0 documentation
+ mlx.core.log10 — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.log1p.html b/docs/build/html/python/_autosummary/mlx.core.log1p.html
index a441cccef5..e3117d7fd1 100644
--- a/docs/build/html/python/_autosummary/mlx.core.log1p.html
+++ b/docs/build/html/python/_autosummary/mlx.core.log1p.html
@@ -8,7 +8,7 @@
mlx.core.log1p — MLX 0.16.0 documentation
+ mlx.core.log1p — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.log2.html b/docs/build/html/python/_autosummary/mlx.core.log2.html
index 0bc572bad7..e9fa8907ce 100644
--- a/docs/build/html/python/_autosummary/mlx.core.log2.html
+++ b/docs/build/html/python/_autosummary/mlx.core.log2.html
@@ -8,7 +8,7 @@
mlx.core.log2 — MLX 0.16.0 documentation
+ mlx.core.log2 — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.logaddexp.html b/docs/build/html/python/_autosummary/mlx.core.logaddexp.html
index 0c761eed32..7527b25c02 100644
--- a/docs/build/html/python/_autosummary/mlx.core.logaddexp.html
+++ b/docs/build/html/python/_autosummary/mlx.core.logaddexp.html
@@ -8,7 +8,7 @@
mlx.core.logaddexp — MLX 0.16.0 documentation
+ mlx.core.logaddexp — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.logical_and.html b/docs/build/html/python/_autosummary/mlx.core.logical_and.html
index c062d03806..3b5a7260ed 100644
--- a/docs/build/html/python/_autosummary/mlx.core.logical_and.html
+++ b/docs/build/html/python/_autosummary/mlx.core.logical_and.html
@@ -8,7 +8,7 @@
mlx.core.logical_and — MLX 0.16.0 documentation
+ mlx.core.logical_and — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.logical_not.html b/docs/build/html/python/_autosummary/mlx.core.logical_not.html
index 5e712c05b3..31c4799287 100644
--- a/docs/build/html/python/_autosummary/mlx.core.logical_not.html
+++ b/docs/build/html/python/_autosummary/mlx.core.logical_not.html
@@ -8,7 +8,7 @@
mlx.core.logical_not — MLX 0.16.0 documentation
+ mlx.core.logical_not — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.logical_or.html b/docs/build/html/python/_autosummary/mlx.core.logical_or.html
index ee365addd9..e68445d693 100644
--- a/docs/build/html/python/_autosummary/mlx.core.logical_or.html
+++ b/docs/build/html/python/_autosummary/mlx.core.logical_or.html
@@ -8,7 +8,7 @@
mlx.core.logical_or — MLX 0.16.0 documentation
+ mlx.core.logical_or — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.logsumexp.html b/docs/build/html/python/_autosummary/mlx.core.logsumexp.html
index 9c782b6a6f..19467af450 100644
--- a/docs/build/html/python/_autosummary/mlx.core.logsumexp.html
+++ b/docs/build/html/python/_autosummary/mlx.core.logsumexp.html
@@ -8,7 +8,7 @@
mlx.core.logsumexp — MLX 0.16.0 documentation
+ mlx.core.logsumexp — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.matmul.html b/docs/build/html/python/_autosummary/mlx.core.matmul.html
index 26107917ca..1e9c1e39db 100644
--- a/docs/build/html/python/_autosummary/mlx.core.matmul.html
+++ b/docs/build/html/python/_autosummary/mlx.core.matmul.html
@@ -8,7 +8,7 @@
mlx.core.matmul — MLX 0.16.0 documentation
+ mlx.core.matmul — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.max.html b/docs/build/html/python/_autosummary/mlx.core.max.html
index db56af0de7..bd9e77e161 100644
--- a/docs/build/html/python/_autosummary/mlx.core.max.html
+++ b/docs/build/html/python/_autosummary/mlx.core.max.html
@@ -8,7 +8,7 @@
mlx.core.max — MLX 0.16.0 documentation
+ mlx.core.max — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.maximum.html b/docs/build/html/python/_autosummary/mlx.core.maximum.html
index f67aac8ba6..13a9880c1b 100644
--- a/docs/build/html/python/_autosummary/mlx.core.maximum.html
+++ b/docs/build/html/python/_autosummary/mlx.core.maximum.html
@@ -8,7 +8,7 @@
mlx.core.maximum — MLX 0.16.0 documentation
+ mlx.core.maximum — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.mean.html b/docs/build/html/python/_autosummary/mlx.core.mean.html
index 3c8258e1e7..6b82fec83e 100644
--- a/docs/build/html/python/_autosummary/mlx.core.mean.html
+++ b/docs/build/html/python/_autosummary/mlx.core.mean.html
@@ -8,7 +8,7 @@
mlx.core.mean — MLX 0.16.0 documentation
+ mlx.core.mean — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.meshgrid.html b/docs/build/html/python/_autosummary/mlx.core.meshgrid.html
index 4fb2f47752..4e61d06eb5 100644
--- a/docs/build/html/python/_autosummary/mlx.core.meshgrid.html
+++ b/docs/build/html/python/_autosummary/mlx.core.meshgrid.html
@@ -8,7 +8,7 @@
mlx.core.meshgrid — MLX 0.16.0 documentation
+ mlx.core.meshgrid — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.metal.clear_cache.html b/docs/build/html/python/_autosummary/mlx.core.metal.clear_cache.html
index c040e604df..f53bed0637 100644
--- a/docs/build/html/python/_autosummary/mlx.core.metal.clear_cache.html
+++ b/docs/build/html/python/_autosummary/mlx.core.metal.clear_cache.html
@@ -8,7 +8,7 @@
mlx.core.metal.clear_cache — MLX 0.16.0 documentation
+ mlx.core.metal.clear_cache — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.metal.device_info.html b/docs/build/html/python/_autosummary/mlx.core.metal.device_info.html
index 78afd77bfb..cdcd971dce 100644
--- a/docs/build/html/python/_autosummary/mlx.core.metal.device_info.html
+++ b/docs/build/html/python/_autosummary/mlx.core.metal.device_info.html
@@ -8,7 +8,7 @@
mlx.core.metal.device_info — MLX 0.16.0 documentation
+ mlx.core.metal.device_info — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.metal.get_active_memory.html b/docs/build/html/python/_autosummary/mlx.core.metal.get_active_memory.html
index f0277df65e..e08e8d56b9 100644
--- a/docs/build/html/python/_autosummary/mlx.core.metal.get_active_memory.html
+++ b/docs/build/html/python/_autosummary/mlx.core.metal.get_active_memory.html
@@ -8,7 +8,7 @@
mlx.core.metal.get_active_memory — MLX 0.16.0 documentation
+ mlx.core.metal.get_active_memory — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.metal.get_cache_memory.html b/docs/build/html/python/_autosummary/mlx.core.metal.get_cache_memory.html
index c3873fcc2c..a3421dd443 100644
--- a/docs/build/html/python/_autosummary/mlx.core.metal.get_cache_memory.html
+++ b/docs/build/html/python/_autosummary/mlx.core.metal.get_cache_memory.html
@@ -8,7 +8,7 @@
mlx.core.metal.get_cache_memory — MLX 0.16.0 documentation
+ mlx.core.metal.get_cache_memory — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.metal.get_peak_memory.html b/docs/build/html/python/_autosummary/mlx.core.metal.get_peak_memory.html
index 891be1daa8..3b24187866 100644
--- a/docs/build/html/python/_autosummary/mlx.core.metal.get_peak_memory.html
+++ b/docs/build/html/python/_autosummary/mlx.core.metal.get_peak_memory.html
@@ -8,7 +8,7 @@
mlx.core.metal.get_peak_memory — MLX 0.16.0 documentation
+ mlx.core.metal.get_peak_memory — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.metal.is_available.html b/docs/build/html/python/_autosummary/mlx.core.metal.is_available.html
index 009bfcf7a9..1740be943b 100644
--- a/docs/build/html/python/_autosummary/mlx.core.metal.is_available.html
+++ b/docs/build/html/python/_autosummary/mlx.core.metal.is_available.html
@@ -8,7 +8,7 @@
mlx.core.metal.is_available — MLX 0.16.0 documentation
+ mlx.core.metal.is_available — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.metal.reset_peak_memory.html b/docs/build/html/python/_autosummary/mlx.core.metal.reset_peak_memory.html
index fefc8c27d5..939a2aa28e 100644
--- a/docs/build/html/python/_autosummary/mlx.core.metal.reset_peak_memory.html
+++ b/docs/build/html/python/_autosummary/mlx.core.metal.reset_peak_memory.html
@@ -8,7 +8,7 @@
mlx.core.metal.reset_peak_memory — MLX 0.16.0 documentation
+ mlx.core.metal.reset_peak_memory — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.metal.set_cache_limit.html b/docs/build/html/python/_autosummary/mlx.core.metal.set_cache_limit.html
index cf40ab2e17..5319360df9 100644
--- a/docs/build/html/python/_autosummary/mlx.core.metal.set_cache_limit.html
+++ b/docs/build/html/python/_autosummary/mlx.core.metal.set_cache_limit.html
@@ -8,7 +8,7 @@
mlx.core.metal.set_cache_limit — MLX 0.16.0 documentation
+ mlx.core.metal.set_cache_limit — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.metal.set_memory_limit.html b/docs/build/html/python/_autosummary/mlx.core.metal.set_memory_limit.html
index 6d58432a2b..06fb95c62d 100644
--- a/docs/build/html/python/_autosummary/mlx.core.metal.set_memory_limit.html
+++ b/docs/build/html/python/_autosummary/mlx.core.metal.set_memory_limit.html
@@ -8,7 +8,7 @@
mlx.core.metal.set_memory_limit — MLX 0.16.0 documentation
+ mlx.core.metal.set_memory_limit — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.metal.start_capture.html b/docs/build/html/python/_autosummary/mlx.core.metal.start_capture.html
index 48ed275929..553dc9c482 100644
--- a/docs/build/html/python/_autosummary/mlx.core.metal.start_capture.html
+++ b/docs/build/html/python/_autosummary/mlx.core.metal.start_capture.html
@@ -8,7 +8,7 @@
mlx.core.metal.start_capture — MLX 0.16.0 documentation
+ mlx.core.metal.start_capture — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.metal.stop_capture.html b/docs/build/html/python/_autosummary/mlx.core.metal.stop_capture.html
index 70ff4ff29e..81467e0266 100644
--- a/docs/build/html/python/_autosummary/mlx.core.metal.stop_capture.html
+++ b/docs/build/html/python/_autosummary/mlx.core.metal.stop_capture.html
@@ -8,7 +8,7 @@
mlx.core.metal.stop_capture — MLX 0.16.0 documentation
+ mlx.core.metal.stop_capture — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.min.html b/docs/build/html/python/_autosummary/mlx.core.min.html
index 513664da48..0d79265c68 100644
--- a/docs/build/html/python/_autosummary/mlx.core.min.html
+++ b/docs/build/html/python/_autosummary/mlx.core.min.html
@@ -8,7 +8,7 @@
mlx.core.min — MLX 0.16.0 documentation
+ mlx.core.min — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.minimum.html b/docs/build/html/python/_autosummary/mlx.core.minimum.html
index 485f74831e..0aa78ae88d 100644
--- a/docs/build/html/python/_autosummary/mlx.core.minimum.html
+++ b/docs/build/html/python/_autosummary/mlx.core.minimum.html
@@ -8,7 +8,7 @@
mlx.core.minimum — MLX 0.16.0 documentation
+ mlx.core.minimum — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.moveaxis.html b/docs/build/html/python/_autosummary/mlx.core.moveaxis.html
index 11753f0771..fab8c49880 100644
--- a/docs/build/html/python/_autosummary/mlx.core.moveaxis.html
+++ b/docs/build/html/python/_autosummary/mlx.core.moveaxis.html
@@ -8,7 +8,7 @@
mlx.core.moveaxis — MLX 0.16.0 documentation
+ mlx.core.moveaxis — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.multiply.html b/docs/build/html/python/_autosummary/mlx.core.multiply.html
index 0c34b8195e..54faffcbcc 100644
--- a/docs/build/html/python/_autosummary/mlx.core.multiply.html
+++ b/docs/build/html/python/_autosummary/mlx.core.multiply.html
@@ -8,7 +8,7 @@
mlx.core.multiply — MLX 0.16.0 documentation
+ mlx.core.multiply — MLX 0.16.1 documentation
@@ -36,14 +36,14 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
@@ -885,11 +889,11 @@ input arrays can also be scalars.
next
-
mlx.core.negative
+
mlx.core.nan_to_num
diff --git a/docs/build/html/python/_autosummary/mlx.core.nan_to_num.html b/docs/build/html/python/_autosummary/mlx.core.nan_to_num.html
new file mode 100644
index 0000000000..047801765a
--- /dev/null
+++ b/docs/build/html/python/_autosummary/mlx.core.nan_to_num.html
@@ -0,0 +1,974 @@
+
+
+
+
+
+
+
+ mlx.core.nan_to_num — MLX 0.16.1 documentation
+
+
+
+
+
+
+ Skip to main content
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
mlx.core.nan_to_num
+
+
+
+
+
+
+
+
+
+
+mlx.core.nan_to_num
+
+
+nan_to_num ( a : scalar | array nan : float = 0 posinf : float | None = None neginf : float | None = None * stream : None | Stream | Device = None ) → array
+Replace NaN and Inf values with finite numbers.
+
+Parameters:
+
+a (array
nan (float , optional ) – Value to replace NaN with. Default: 0
posinf (float , optional ) – Value to replace positive infinities
+with. If None None
neginf (float , optional ) – Value to replace negative infinities
+with. If None None
+Returns:
+Output array with NaN and Inf replaced.
+Return type:
+array
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
mlx.core.negative — MLX 0.16.0 documentation
+ mlx.core.negative — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.not_equal.html b/docs/build/html/python/_autosummary/mlx.core.not_equal.html
index a11088388a..7089fee9c7 100644
--- a/docs/build/html/python/_autosummary/mlx.core.not_equal.html
+++ b/docs/build/html/python/_autosummary/mlx.core.not_equal.html
@@ -8,7 +8,7 @@
mlx.core.not_equal — MLX 0.16.0 documentation
+ mlx.core.not_equal — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.ones.html b/docs/build/html/python/_autosummary/mlx.core.ones.html
index 28b91a1fbd..0363830dcd 100644
--- a/docs/build/html/python/_autosummary/mlx.core.ones.html
+++ b/docs/build/html/python/_autosummary/mlx.core.ones.html
@@ -8,7 +8,7 @@
mlx.core.ones — MLX 0.16.0 documentation
+ mlx.core.ones — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.ones_like.html b/docs/build/html/python/_autosummary/mlx.core.ones_like.html
index c5fc695408..69340a1b21 100644
--- a/docs/build/html/python/_autosummary/mlx.core.ones_like.html
+++ b/docs/build/html/python/_autosummary/mlx.core.ones_like.html
@@ -8,7 +8,7 @@
mlx.core.ones_like — MLX 0.16.0 documentation
+ mlx.core.ones_like — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.outer.html b/docs/build/html/python/_autosummary/mlx.core.outer.html
index afca0a63c0..98d9d0ccb3 100644
--- a/docs/build/html/python/_autosummary/mlx.core.outer.html
+++ b/docs/build/html/python/_autosummary/mlx.core.outer.html
@@ -8,7 +8,7 @@
mlx.core.outer — MLX 0.16.0 documentation
+ mlx.core.outer — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.pad.html b/docs/build/html/python/_autosummary/mlx.core.pad.html
index c7b10baac5..bb0ebb315a 100644
--- a/docs/build/html/python/_autosummary/mlx.core.pad.html
+++ b/docs/build/html/python/_autosummary/mlx.core.pad.html
@@ -8,7 +8,7 @@
mlx.core.pad — MLX 0.16.0 documentation
+ mlx.core.pad — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
@@ -842,7 +846,7 @@ document.write(`
mlx.core.pad
-pad ( a : array pad_with : int | Tuple [ int ] | Tuple [ int , int ] | List [ Tuple [ int , int ] ] constant_values : scalar | array = 0 * stream : None | Stream | Device = None ) → array
+pad ( a : array pad_width : int | Tuple [ int ] | Tuple [ int , int ] | List [ Tuple [ int , int ] ] constant_values : scalar | array = 0 * stream : None | Stream | Device = None ) → array Pad an array with a constant value
Parameters:
diff --git a/docs/build/html/python/_autosummary/mlx.core.partition.html b/docs/build/html/python/_autosummary/mlx.core.partition.html
index 439bbe0ffc..966c1904f2 100644
--- a/docs/build/html/python/_autosummary/mlx.core.partition.html
+++ b/docs/build/html/python/_autosummary/mlx.core.partition.html
@@ -8,7 +8,7 @@
mlx.core.partition — MLX 0.16.0 documentation
+ mlx.core.partition — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.power.html b/docs/build/html/python/_autosummary/mlx.core.power.html
index b01a71d986..d5873250d3 100644
--- a/docs/build/html/python/_autosummary/mlx.core.power.html
+++ b/docs/build/html/python/_autosummary/mlx.core.power.html
@@ -8,7 +8,7 @@
mlx.core.power — MLX 0.16.0 documentation
+ mlx.core.power — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.prod.html b/docs/build/html/python/_autosummary/mlx.core.prod.html
index 65e78acf06..c933eddc67 100644
--- a/docs/build/html/python/_autosummary/mlx.core.prod.html
+++ b/docs/build/html/python/_autosummary/mlx.core.prod.html
@@ -8,7 +8,7 @@
mlx.core.prod — MLX 0.16.0 documentation
+ mlx.core.prod — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.quantize.html b/docs/build/html/python/_autosummary/mlx.core.quantize.html
index fb92e1fa5c..3f45531158 100644
--- a/docs/build/html/python/_autosummary/mlx.core.quantize.html
+++ b/docs/build/html/python/_autosummary/mlx.core.quantize.html
@@ -8,7 +8,7 @@
mlx.core.quantize — MLX 0.16.0 documentation
+ mlx.core.quantize — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
@@ -876,9 +880,9 @@ save \(s\) and
w (array
group_size (int , optional ) – The size of the group in w 64
64 bits (int , optional ) – The number of bits occupied by each element of
-w 4
w 4 Returns:
diff --git a/docs/build/html/python/_autosummary/mlx.core.quantized_matmul.html b/docs/build/html/python/_autosummary/mlx.core.quantized_matmul.html
index 4596fca46d..eee8bf0a2e 100644
--- a/docs/build/html/python/_autosummary/mlx.core.quantized_matmul.html
+++ b/docs/build/html/python/_autosummary/mlx.core.quantized_matmul.html
@@ -8,7 +8,7 @@
mlx.core.quantized_matmul — MLX 0.16.0 documentation
+ mlx.core.quantized_matmul — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace Transforms
@@ -856,11 +860,11 @@ unsigned 32 bit integer.
biases (array group_size w
transpose (bool , optional ) – Defines whether to multiply with the
transposed w x @ w.T x @ w True
x @ w.T x @ w True group_size (int , optional ) – The size of the group in w 64
64 bits (int , optional ) – The number of bits occupied by each element in
-w 4
w 4 Returns:
diff --git a/docs/build/html/python/_autosummary/mlx.core.radians.html b/docs/build/html/python/_autosummary/mlx.core.radians.html
index 9d575a10d4..6b1e3aff8a 100644
--- a/docs/build/html/python/_autosummary/mlx.core.radians.html
+++ b/docs/build/html/python/_autosummary/mlx.core.radians.html
@@ -8,7 +8,7 @@
mlx.core.radians — MLX 0.16.0 documentation
+ mlx.core.radians — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.random.bernoulli.html b/docs/build/html/python/_autosummary/mlx.core.random.bernoulli.html
index 96a787de3d..5dc0aefb1e 100644
--- a/docs/build/html/python/_autosummary/mlx.core.random.bernoulli.html
+++ b/docs/build/html/python/_autosummary/mlx.core.random.bernoulli.html
@@ -8,7 +8,7 @@
mlx.core.random.bernoulli — MLX 0.16.0 documentation
+ mlx.core.random.bernoulli — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
@@ -851,10 +855,10 @@ must be broadcastable to Parameters:
p (float or array , optional ) – Parameter of the Bernoulli
-distribution. Default is 0.5.
shape (list ( int ) , optional ) – Shape of the output. The default
-shape is p.shape
key (array , optional ) – A PRNG key. Default: None.
0.5 shape (list ( int ) , optional ) – Shape of the output.
+Default: p.shape
key (array , optional ) – A PRNG key. Default: None
Returns:
diff --git a/docs/build/html/python/_autosummary/mlx.core.random.categorical.html b/docs/build/html/python/_autosummary/mlx.core.random.categorical.html
index dc5a5cdba0..269220e587 100644
--- a/docs/build/html/python/_autosummary/mlx.core.random.categorical.html
+++ b/docs/build/html/python/_autosummary/mlx.core.random.categorical.html
@@ -8,7 +8,7 @@
mlx.core.random.categorical — MLX 0.16.0 documentation
+ mlx.core.random.categorical — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace Transforms
@@ -853,14 +857,14 @@ has the same shape as
logits (array unnormalized categorical distribution(s).
axis (int , optional ) – The axis which specifies the distribution.
-Default is -1
-1 shape (list ( int ) , optional ) – The shape of the output. This must
be broadcast compatable with logits.shape axis None
num_samples (int , optional ) – The number of samples to draw from each
of the categorical distributions in logits num_samples None
key (array , optional ) – A PRNG key. Default: None.
key (array , optional ) – A PRNG key. Default: None
Returns:
diff --git a/docs/build/html/python/_autosummary/mlx.core.random.gumbel.html b/docs/build/html/python/_autosummary/mlx.core.random.gumbel.html
index 699811ef96..c63c24c728 100644
--- a/docs/build/html/python/_autosummary/mlx.core.random.gumbel.html
+++ b/docs/build/html/python/_autosummary/mlx.core.random.gumbel.html
@@ -8,7 +8,7 @@
mlx.core.random.gumbel — MLX 0.16.0 documentation
+ mlx.core.random.gumbel — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace Transforms
@@ -850,7 +854,7 @@ which CDF exp(-exp(
Parameters:
shape (list ( int ) ) – The shape of the output.
key (array , optional ) – A PRNG key. Default: None.
key (array , optional ) – A PRNG key. Default: None
Returns:
diff --git a/docs/build/html/python/_autosummary/mlx.core.random.key.html b/docs/build/html/python/_autosummary/mlx.core.random.key.html
index a7815c5a61..2c1832af69 100644
--- a/docs/build/html/python/_autosummary/mlx.core.random.key.html
+++ b/docs/build/html/python/_autosummary/mlx.core.random.key.html
@@ -8,7 +8,7 @@
mlx.core.random.key — MLX 0.16.0 documentation
+ mlx.core.random.key — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.random.laplace.html b/docs/build/html/python/_autosummary/mlx.core.random.laplace.html
new file mode 100644
index 0000000000..ce6b7ae408
--- /dev/null
+++ b/docs/build/html/python/_autosummary/mlx.core.random.laplace.html
@@ -0,0 +1,972 @@
+
+
+
+
+
+
+
+ mlx.core.random.laplace — MLX 0.16.1 documentation
+
+
+
+
+
+
+ Skip to main content
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
mlx.core.random.laplace
+
+
+
+
+
+
+
+
+
+
+mlx.core.random.laplace
+
+
+laplace ( shape : Sequence [ int ] = [] dtype : Dtype | None = float32 loc : float = 0.0 scale : float = 1.0 key : array | None = None stream : None | Stream | Device = None ) → array
+Sample numbers from a Laplace distribution.
+
+Parameters:
+
+shape (list ( int ) , optional ) – Shape of the output. Default: ()
dtype (Dtype , optional ) – Type of the output. Default: float32
loc (float , optional ) – Mean of the distribution. Default: 0.0
scale (float , optional ) – The scale “b” of the Laplace distribution.
+Default:1.0
key (array , optional ) – A PRNG key. Default: None
+Returns:
+The output array of random values.
+Return type:
+array
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
mlx.core.random.multivariate_normal — MLX 0.16.0 documentation
+ mlx.core.random.multivariate_normal — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.random.normal.html b/docs/build/html/python/_autosummary/mlx.core.random.normal.html
index f70d04bad1..e30c9b9b7a 100644
--- a/docs/build/html/python/_autosummary/mlx.core.random.normal.html
+++ b/docs/build/html/python/_autosummary/mlx.core.random.normal.html
@@ -8,7 +8,7 @@
mlx.core.random.normal — MLX 0.16.0 documentation
+ mlx.core.random.normal — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.random.randint.html b/docs/build/html/python/_autosummary/mlx.core.random.randint.html
index 2f60543fb2..31bea48547 100644
--- a/docs/build/html/python/_autosummary/mlx.core.random.randint.html
+++ b/docs/build/html/python/_autosummary/mlx.core.random.randint.html
@@ -8,7 +8,7 @@
mlx.core.random.randint — MLX 0.16.0 documentation
+ mlx.core.random.randint — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
@@ -852,9 +856,9 @@ scalars or arrays and must be roadcastable to
low (scalar or array
high (scalar or array
shape (list ( int ) , optional ) – Shape of the output. Defaults to ()
dtype (Dtype , optional ) – Type of the output. Defaults to int32
key (array , optional ) – A PRNG key. Default: None.
shape (list ( int ) , optional ) – Shape of the output. Default: ()
dtype (Dtype , optional ) – Type of the output. Default: int32
key (array , optional ) – A PRNG key. Default: None
Returns:
diff --git a/docs/build/html/python/_autosummary/mlx.core.random.seed.html b/docs/build/html/python/_autosummary/mlx.core.random.seed.html
index 5d0d923c21..28fff028d9 100644
--- a/docs/build/html/python/_autosummary/mlx.core.random.seed.html
+++ b/docs/build/html/python/_autosummary/mlx.core.random.seed.html
@@ -8,7 +8,7 @@
mlx.core.random.seed — MLX 0.16.0 documentation
+ mlx.core.random.seed — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.random.split.html b/docs/build/html/python/_autosummary/mlx.core.random.split.html
index 73fb23cd8e..e8d4143568 100644
--- a/docs/build/html/python/_autosummary/mlx.core.random.split.html
+++ b/docs/build/html/python/_autosummary/mlx.core.random.split.html
@@ -8,7 +8,7 @@
mlx.core.random.split — MLX 0.16.0 documentation
+ mlx.core.random.split — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
@@ -848,7 +852,7 @@ document.write(`
Parameters:
key (array
num (int , optional ) – Number of sub keys. Default is 2.
num (int , optional ) – Number of sub keys. Default: 2
Returns:
diff --git a/docs/build/html/python/_autosummary/mlx.core.random.truncated_normal.html b/docs/build/html/python/_autosummary/mlx.core.random.truncated_normal.html
index 67d3c25071..70190e1117 100644
--- a/docs/build/html/python/_autosummary/mlx.core.random.truncated_normal.html
+++ b/docs/build/html/python/_autosummary/mlx.core.random.truncated_normal.html
@@ -8,7 +8,7 @@
mlx.core.random.truncated_normal — MLX 0.16.0 documentation
+ mlx.core.random.truncated_normal — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
@@ -853,10 +857,10 @@ can be scalars or arrays and must be broadcastable to array ) – Lower bound of the domain. upper (scalar or array
shape (list ( int ) , optional ) – The shape of the output.
-Default is ()
() dtype (Dtype , optional ) – The data type of the output.
-Default is float32
key (array , optional ) – A PRNG key. Default: None.
float32 key (array , optional ) – A PRNG key. Default: None
Returns:
diff --git a/docs/build/html/python/_autosummary/mlx.core.random.uniform.html b/docs/build/html/python/_autosummary/mlx.core.random.uniform.html
index 3b70253e00..846475fe45 100644
--- a/docs/build/html/python/_autosummary/mlx.core.random.uniform.html
+++ b/docs/build/html/python/_autosummary/mlx.core.random.uniform.html
@@ -8,7 +8,7 @@
mlx.core.random.uniform — MLX 0.16.0 documentation
+ mlx.core.random.uniform — MLX 0.16.1 documentation
@@ -36,14 +36,14 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace Transforms
@@ -850,11 +854,13 @@ broadcastable to sh
Parameters:
-low (scalar or array , optional ) – Lower bound of the distribution. Default is 0
high (scalar or array , optional ) – Upper bound of the distribution. Default is 1
shape (list ( int ) , optional ) – Shape of the output. Default is ()
low (scalar or array , optional ) – Lower bound of the distribution.
+Default: 0
high (scalar or array , optional ) – Upper bound of the distribution.
+Default: 1
shape (list ( int ) , optional ) – Shape of the output. Default:()
dtype (Dtype , optional ) – Type of the output. Default: float32
key (array , optional ) – A PRNG key. Default: None
dtype (Dtype , optional ) – Type of the output. Default is float32
Returns:
@@ -889,11 +895,11 @@ broadcastable to sh
next
-
Transforms
+
mlx.core.random.laplace
diff --git a/docs/build/html/python/_autosummary/mlx.core.reciprocal.html b/docs/build/html/python/_autosummary/mlx.core.reciprocal.html
index c90ad56f0d..5edee69d83 100644
--- a/docs/build/html/python/_autosummary/mlx.core.reciprocal.html
+++ b/docs/build/html/python/_autosummary/mlx.core.reciprocal.html
@@ -8,7 +8,7 @@
mlx.core.reciprocal — MLX 0.16.0 documentation
+ mlx.core.reciprocal — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.remainder.html b/docs/build/html/python/_autosummary/mlx.core.remainder.html
index 13d76a6064..8873222fff 100644
--- a/docs/build/html/python/_autosummary/mlx.core.remainder.html
+++ b/docs/build/html/python/_autosummary/mlx.core.remainder.html
@@ -8,7 +8,7 @@
mlx.core.remainder — MLX 0.16.0 documentation
+ mlx.core.remainder — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.repeat.html b/docs/build/html/python/_autosummary/mlx.core.repeat.html
index 32ef57124e..b8af5997ed 100644
--- a/docs/build/html/python/_autosummary/mlx.core.repeat.html
+++ b/docs/build/html/python/_autosummary/mlx.core.repeat.html
@@ -8,7 +8,7 @@
mlx.core.repeat — MLX 0.16.0 documentation
+ mlx.core.repeat — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.reshape.html b/docs/build/html/python/_autosummary/mlx.core.reshape.html
index 96eb6702a4..3331984d23 100644
--- a/docs/build/html/python/_autosummary/mlx.core.reshape.html
+++ b/docs/build/html/python/_autosummary/mlx.core.reshape.html
@@ -8,7 +8,7 @@
mlx.core.reshape — MLX 0.16.0 documentation
+ mlx.core.reshape — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.right_shift.html b/docs/build/html/python/_autosummary/mlx.core.right_shift.html
index 27fe938d75..987bb3d56d 100644
--- a/docs/build/html/python/_autosummary/mlx.core.right_shift.html
+++ b/docs/build/html/python/_autosummary/mlx.core.right_shift.html
@@ -8,7 +8,7 @@
mlx.core.right_shift — MLX 0.16.0 documentation
+ mlx.core.right_shift — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.round.html b/docs/build/html/python/_autosummary/mlx.core.round.html
index b3a334a356..0c3a24901b 100644
--- a/docs/build/html/python/_autosummary/mlx.core.round.html
+++ b/docs/build/html/python/_autosummary/mlx.core.round.html
@@ -8,7 +8,7 @@
mlx.core.round — MLX 0.16.0 documentation
+ mlx.core.round — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.rsqrt.html b/docs/build/html/python/_autosummary/mlx.core.rsqrt.html
index fba34f0238..6e016a10ea 100644
--- a/docs/build/html/python/_autosummary/mlx.core.rsqrt.html
+++ b/docs/build/html/python/_autosummary/mlx.core.rsqrt.html
@@ -8,7 +8,7 @@
mlx.core.rsqrt — MLX 0.16.0 documentation
+ mlx.core.rsqrt — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.save.html b/docs/build/html/python/_autosummary/mlx.core.save.html
index db80101865..ce39c2d29c 100644
--- a/docs/build/html/python/_autosummary/mlx.core.save.html
+++ b/docs/build/html/python/_autosummary/mlx.core.save.html
@@ -8,7 +8,7 @@
mlx.core.save — MLX 0.16.0 documentation
+ mlx.core.save — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.save_gguf.html b/docs/build/html/python/_autosummary/mlx.core.save_gguf.html
index b2735d749c..e6f18aa61e 100644
--- a/docs/build/html/python/_autosummary/mlx.core.save_gguf.html
+++ b/docs/build/html/python/_autosummary/mlx.core.save_gguf.html
@@ -8,7 +8,7 @@
mlx.core.save_gguf — MLX 0.16.0 documentation
+ mlx.core.save_gguf — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
@@ -850,9 +854,10 @@ more information on the format.
Parameters:
file (file , str
arrays (dict ( str , array ) ) – The dictionary of names to arrays to
metadata (be saved. ) –
of (The dictionary ) – metadata to be saved. The values can be a scalar or 1D
+
arrays (dict ( str , array ) ) – The dictionary of names to arrays to
+be saved.
metadata (dict ( str , Union [ array , str , list ( str ) ] ) ) – The dictionary
+of metadata to be saved. The values can be a scalar or 1D
obj:array , a str list str
mlx.core.save_safetensors — MLX 0.16.0 documentation
+ mlx.core.save_safetensors — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.savez.html b/docs/build/html/python/_autosummary/mlx.core.savez.html
index 2992c2d81d..5d9973b578 100644
--- a/docs/build/html/python/_autosummary/mlx.core.savez.html
+++ b/docs/build/html/python/_autosummary/mlx.core.savez.html
@@ -8,7 +8,7 @@
mlx.core.savez — MLX 0.16.0 documentation
+ mlx.core.savez — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.savez_compressed.html b/docs/build/html/python/_autosummary/mlx.core.savez_compressed.html
index c58c47b67d..16fccc69e9 100644
--- a/docs/build/html/python/_autosummary/mlx.core.savez_compressed.html
+++ b/docs/build/html/python/_autosummary/mlx.core.savez_compressed.html
@@ -8,7 +8,7 @@
mlx.core.savez_compressed — MLX 0.16.0 documentation
+ mlx.core.savez_compressed — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.set_default_device.html b/docs/build/html/python/_autosummary/mlx.core.set_default_device.html
index 08d9304872..795ba4610a 100644
--- a/docs/build/html/python/_autosummary/mlx.core.set_default_device.html
+++ b/docs/build/html/python/_autosummary/mlx.core.set_default_device.html
@@ -8,7 +8,7 @@
mlx.core.set_default_device — MLX 0.16.0 documentation
+ mlx.core.set_default_device — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.set_default_stream.html b/docs/build/html/python/_autosummary/mlx.core.set_default_stream.html
index 07de931a41..64c9be57d5 100644
--- a/docs/build/html/python/_autosummary/mlx.core.set_default_stream.html
+++ b/docs/build/html/python/_autosummary/mlx.core.set_default_stream.html
@@ -8,7 +8,7 @@
mlx.core.set_default_stream — MLX 0.16.0 documentation
+ mlx.core.set_default_stream — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.sigmoid.html b/docs/build/html/python/_autosummary/mlx.core.sigmoid.html
index e5c61c6f84..cb5587dd27 100644
--- a/docs/build/html/python/_autosummary/mlx.core.sigmoid.html
+++ b/docs/build/html/python/_autosummary/mlx.core.sigmoid.html
@@ -8,7 +8,7 @@
mlx.core.sigmoid — MLX 0.16.0 documentation
+ mlx.core.sigmoid — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.sign.html b/docs/build/html/python/_autosummary/mlx.core.sign.html
index 7493be1613..d48df9a3c9 100644
--- a/docs/build/html/python/_autosummary/mlx.core.sign.html
+++ b/docs/build/html/python/_autosummary/mlx.core.sign.html
@@ -8,7 +8,7 @@
mlx.core.sign — MLX 0.16.0 documentation
+ mlx.core.sign — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.sin.html b/docs/build/html/python/_autosummary/mlx.core.sin.html
index 4ec0226a44..e179809cf0 100644
--- a/docs/build/html/python/_autosummary/mlx.core.sin.html
+++ b/docs/build/html/python/_autosummary/mlx.core.sin.html
@@ -8,7 +8,7 @@
mlx.core.sin — MLX 0.16.0 documentation
+ mlx.core.sin — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.sinh.html b/docs/build/html/python/_autosummary/mlx.core.sinh.html
index 735b6adf55..f83441c7e2 100644
--- a/docs/build/html/python/_autosummary/mlx.core.sinh.html
+++ b/docs/build/html/python/_autosummary/mlx.core.sinh.html
@@ -8,7 +8,7 @@
mlx.core.sinh — MLX 0.16.0 documentation
+ mlx.core.sinh — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.softmax.html b/docs/build/html/python/_autosummary/mlx.core.softmax.html
index a4b6744bbc..a71532717f 100644
--- a/docs/build/html/python/_autosummary/mlx.core.softmax.html
+++ b/docs/build/html/python/_autosummary/mlx.core.softmax.html
@@ -8,7 +8,7 @@
mlx.core.softmax — MLX 0.16.0 documentation
+ mlx.core.softmax — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.sort.html b/docs/build/html/python/_autosummary/mlx.core.sort.html
index 59eb311242..b7cf82e1b2 100644
--- a/docs/build/html/python/_autosummary/mlx.core.sort.html
+++ b/docs/build/html/python/_autosummary/mlx.core.sort.html
@@ -8,7 +8,7 @@
mlx.core.sort — MLX 0.16.0 documentation
+ mlx.core.sort — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.split.html b/docs/build/html/python/_autosummary/mlx.core.split.html
index 61e60335be..a6ce389003 100644
--- a/docs/build/html/python/_autosummary/mlx.core.split.html
+++ b/docs/build/html/python/_autosummary/mlx.core.split.html
@@ -8,7 +8,7 @@
mlx.core.split — MLX 0.16.0 documentation
+ mlx.core.split — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.sqrt.html b/docs/build/html/python/_autosummary/mlx.core.sqrt.html
index 6af88f8b1f..5741288a06 100644
--- a/docs/build/html/python/_autosummary/mlx.core.sqrt.html
+++ b/docs/build/html/python/_autosummary/mlx.core.sqrt.html
@@ -8,7 +8,7 @@
mlx.core.sqrt — MLX 0.16.0 documentation
+ mlx.core.sqrt — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.square.html b/docs/build/html/python/_autosummary/mlx.core.square.html
index 84a6677a88..76f1a44c44 100644
--- a/docs/build/html/python/_autosummary/mlx.core.square.html
+++ b/docs/build/html/python/_autosummary/mlx.core.square.html
@@ -8,7 +8,7 @@
mlx.core.square — MLX 0.16.0 documentation
+ mlx.core.square — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.squeeze.html b/docs/build/html/python/_autosummary/mlx.core.squeeze.html
index ca0188c71f..88fc1d71a2 100644
--- a/docs/build/html/python/_autosummary/mlx.core.squeeze.html
+++ b/docs/build/html/python/_autosummary/mlx.core.squeeze.html
@@ -8,7 +8,7 @@
mlx.core.squeeze — MLX 0.16.0 documentation
+ mlx.core.squeeze — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.stack.html b/docs/build/html/python/_autosummary/mlx.core.stack.html
index 4b36f58c19..96ac5fc6c7 100644
--- a/docs/build/html/python/_autosummary/mlx.core.stack.html
+++ b/docs/build/html/python/_autosummary/mlx.core.stack.html
@@ -8,7 +8,7 @@
mlx.core.stack — MLX 0.16.0 documentation
+ mlx.core.stack — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.std.html b/docs/build/html/python/_autosummary/mlx.core.std.html
index 9153e0861f..0e1ee56263 100644
--- a/docs/build/html/python/_autosummary/mlx.core.std.html
+++ b/docs/build/html/python/_autosummary/mlx.core.std.html
@@ -8,7 +8,7 @@
mlx.core.std — MLX 0.16.0 documentation
+ mlx.core.std — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.stop_gradient.html b/docs/build/html/python/_autosummary/mlx.core.stop_gradient.html
index cd670ee2e5..c6fc13ce23 100644
--- a/docs/build/html/python/_autosummary/mlx.core.stop_gradient.html
+++ b/docs/build/html/python/_autosummary/mlx.core.stop_gradient.html
@@ -8,7 +8,7 @@
mlx.core.stop_gradient — MLX 0.16.0 documentation
+ mlx.core.stop_gradient — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
@@ -851,11 +855,8 @@ through the array.
a (array
Returns:
-
-The unchanged input a through it.
-
-
+The unchanged input a
Return type:
array
diff --git a/docs/build/html/python/_autosummary/mlx.core.subtract.html b/docs/build/html/python/_autosummary/mlx.core.subtract.html
index fe5480ea96..340c24ae1d 100644
--- a/docs/build/html/python/_autosummary/mlx.core.subtract.html
+++ b/docs/build/html/python/_autosummary/mlx.core.subtract.html
@@ -8,7 +8,7 @@
mlx.core.subtract — MLX 0.16.0 documentation
+ mlx.core.subtract — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.sum.html b/docs/build/html/python/_autosummary/mlx.core.sum.html
index bd158b0fc3..76cfba25af 100644
--- a/docs/build/html/python/_autosummary/mlx.core.sum.html
+++ b/docs/build/html/python/_autosummary/mlx.core.sum.html
@@ -8,7 +8,7 @@
mlx.core.sum — MLX 0.16.0 documentation
+ mlx.core.sum — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.swapaxes.html b/docs/build/html/python/_autosummary/mlx.core.swapaxes.html
index 519df1bcf8..1d27ba1563 100644
--- a/docs/build/html/python/_autosummary/mlx.core.swapaxes.html
+++ b/docs/build/html/python/_autosummary/mlx.core.swapaxes.html
@@ -8,7 +8,7 @@
mlx.core.swapaxes — MLX 0.16.0 documentation
+ mlx.core.swapaxes — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.synchronize.html b/docs/build/html/python/_autosummary/mlx.core.synchronize.html
index bdebd603d4..638e669066 100644
--- a/docs/build/html/python/_autosummary/mlx.core.synchronize.html
+++ b/docs/build/html/python/_autosummary/mlx.core.synchronize.html
@@ -8,7 +8,7 @@
mlx.core.synchronize — MLX 0.16.0 documentation
+ mlx.core.synchronize — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.take.html b/docs/build/html/python/_autosummary/mlx.core.take.html
index 65ff4f10ed..f1a294edde 100644
--- a/docs/build/html/python/_autosummary/mlx.core.take.html
+++ b/docs/build/html/python/_autosummary/mlx.core.take.html
@@ -8,7 +8,7 @@
mlx.core.take — MLX 0.16.0 documentation
+ mlx.core.take — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.take_along_axis.html b/docs/build/html/python/_autosummary/mlx.core.take_along_axis.html
index ecce634eac..d83d7c52a0 100644
--- a/docs/build/html/python/_autosummary/mlx.core.take_along_axis.html
+++ b/docs/build/html/python/_autosummary/mlx.core.take_along_axis.html
@@ -8,7 +8,7 @@
mlx.core.take_along_axis — MLX 0.16.0 documentation
+ mlx.core.take_along_axis — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.tan.html b/docs/build/html/python/_autosummary/mlx.core.tan.html
index 2f5aa7866d..35ad16308e 100644
--- a/docs/build/html/python/_autosummary/mlx.core.tan.html
+++ b/docs/build/html/python/_autosummary/mlx.core.tan.html
@@ -8,7 +8,7 @@
mlx.core.tan — MLX 0.16.0 documentation
+ mlx.core.tan — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.tanh.html b/docs/build/html/python/_autosummary/mlx.core.tanh.html
index 6f6fcb924a..d301b1b05d 100644
--- a/docs/build/html/python/_autosummary/mlx.core.tanh.html
+++ b/docs/build/html/python/_autosummary/mlx.core.tanh.html
@@ -8,7 +8,7 @@
mlx.core.tanh — MLX 0.16.0 documentation
+ mlx.core.tanh — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.tensordot.html b/docs/build/html/python/_autosummary/mlx.core.tensordot.html
index 7ef6db273f..b63a8dafbf 100644
--- a/docs/build/html/python/_autosummary/mlx.core.tensordot.html
+++ b/docs/build/html/python/_autosummary/mlx.core.tensordot.html
@@ -8,7 +8,7 @@
mlx.core.tensordot — MLX 0.16.0 documentation
+ mlx.core.tensordot — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
@@ -853,7 +857,7 @@ document.write(`
sum over. If an integer is provided, then sum over the last
axes a axes b a b a b Returns:
diff --git a/docs/build/html/python/_autosummary/mlx.core.tile.html b/docs/build/html/python/_autosummary/mlx.core.tile.html
index 13371e0d57..f813b6efea 100644
--- a/docs/build/html/python/_autosummary/mlx.core.tile.html
+++ b/docs/build/html/python/_autosummary/mlx.core.tile.html
@@ -8,7 +8,7 @@
mlx.core.tile — MLX 0.16.0 documentation
+ mlx.core.tile — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.topk.html b/docs/build/html/python/_autosummary/mlx.core.topk.html
index fbba22c635..204da0eebb 100644
--- a/docs/build/html/python/_autosummary/mlx.core.topk.html
+++ b/docs/build/html/python/_autosummary/mlx.core.topk.html
@@ -8,7 +8,7 @@
mlx.core.topk — MLX 0.16.0 documentation
+ mlx.core.topk — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.trace.html b/docs/build/html/python/_autosummary/mlx.core.trace.html
index a310c3dbe6..96a67b873c 100644
--- a/docs/build/html/python/_autosummary/mlx.core.trace.html
+++ b/docs/build/html/python/_autosummary/mlx.core.trace.html
@@ -8,7 +8,7 @@
mlx.core.trace — MLX 0.16.0 documentation
+ mlx.core.trace — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.transpose.html b/docs/build/html/python/_autosummary/mlx.core.transpose.html
index b6e3b64256..0ad65fd084 100644
--- a/docs/build/html/python/_autosummary/mlx.core.transpose.html
+++ b/docs/build/html/python/_autosummary/mlx.core.transpose.html
@@ -8,7 +8,7 @@
mlx.core.transpose — MLX 0.16.0 documentation
+ mlx.core.transpose — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.tri.html b/docs/build/html/python/_autosummary/mlx.core.tri.html
index 0f67ec5151..292a4eab65 100644
--- a/docs/build/html/python/_autosummary/mlx.core.tri.html
+++ b/docs/build/html/python/_autosummary/mlx.core.tri.html
@@ -8,7 +8,7 @@
mlx.core.tri — MLX 0.16.0 documentation
+ mlx.core.tri — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.tril.html b/docs/build/html/python/_autosummary/mlx.core.tril.html
index 8f84edc007..5fa057f2c8 100644
--- a/docs/build/html/python/_autosummary/mlx.core.tril.html
+++ b/docs/build/html/python/_autosummary/mlx.core.tril.html
@@ -8,7 +8,7 @@
mlx.core.tril — MLX 0.16.0 documentation
+ mlx.core.tril — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.triu.html b/docs/build/html/python/_autosummary/mlx.core.triu.html
index 47e290e900..16f0ef3b0a 100644
--- a/docs/build/html/python/_autosummary/mlx.core.triu.html
+++ b/docs/build/html/python/_autosummary/mlx.core.triu.html
@@ -8,7 +8,7 @@
mlx.core.triu — MLX 0.16.0 documentation
+ mlx.core.triu — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.value_and_grad.html b/docs/build/html/python/_autosummary/mlx.core.value_and_grad.html
index 04bd02fbb0..503887fcc3 100644
--- a/docs/build/html/python/_autosummary/mlx.core.value_and_grad.html
+++ b/docs/build/html/python/_autosummary/mlx.core.value_and_grad.html
@@ -8,7 +8,7 @@
mlx.core.value_and_grad — MLX 0.16.0 documentation
+ mlx.core.value_and_grad — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.var.html b/docs/build/html/python/_autosummary/mlx.core.var.html
index 880319b32f..5797233a11 100644
--- a/docs/build/html/python/_autosummary/mlx.core.var.html
+++ b/docs/build/html/python/_autosummary/mlx.core.var.html
@@ -8,7 +8,7 @@
mlx.core.var — MLX 0.16.0 documentation
+ mlx.core.var — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.view.html b/docs/build/html/python/_autosummary/mlx.core.view.html
index def1ef307f..0f1ce84226 100644
--- a/docs/build/html/python/_autosummary/mlx.core.view.html
+++ b/docs/build/html/python/_autosummary/mlx.core.view.html
@@ -8,7 +8,7 @@
mlx.core.view — MLX 0.16.0 documentation
+ mlx.core.view — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.vjp.html b/docs/build/html/python/_autosummary/mlx.core.vjp.html
index 75f41da6a6..2b07105690 100644
--- a/docs/build/html/python/_autosummary/mlx.core.vjp.html
+++ b/docs/build/html/python/_autosummary/mlx.core.vjp.html
@@ -8,7 +8,7 @@
mlx.core.vjp — MLX 0.16.0 documentation
+ mlx.core.vjp — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.vmap.html b/docs/build/html/python/_autosummary/mlx.core.vmap.html
index 2dcfed998b..adc4890b4c 100644
--- a/docs/build/html/python/_autosummary/mlx.core.vmap.html
+++ b/docs/build/html/python/_autosummary/mlx.core.vmap.html
@@ -8,7 +8,7 @@
mlx.core.vmap — MLX 0.16.0 documentation
+ mlx.core.vmap — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.where.html b/docs/build/html/python/_autosummary/mlx.core.where.html
index 9ad1a32019..5a835b047a 100644
--- a/docs/build/html/python/_autosummary/mlx.core.where.html
+++ b/docs/build/html/python/_autosummary/mlx.core.where.html
@@ -8,7 +8,7 @@
mlx.core.where — MLX 0.16.0 documentation
+ mlx.core.where — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.zeros.html b/docs/build/html/python/_autosummary/mlx.core.zeros.html
index a68d69198f..d5a05d21bd 100644
--- a/docs/build/html/python/_autosummary/mlx.core.zeros.html
+++ b/docs/build/html/python/_autosummary/mlx.core.zeros.html
@@ -8,7 +8,7 @@
mlx.core.zeros — MLX 0.16.0 documentation
+ mlx.core.zeros — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.core.zeros_like.html b/docs/build/html/python/_autosummary/mlx.core.zeros_like.html
index 627f78b334..585d02447c 100644
--- a/docs/build/html/python/_autosummary/mlx.core.zeros_like.html
+++ b/docs/build/html/python/_autosummary/mlx.core.zeros_like.html
@@ -8,7 +8,7 @@
mlx.core.zeros_like — MLX 0.16.0 documentation
+ mlx.core.zeros_like — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.nn.quantize.html b/docs/build/html/python/_autosummary/mlx.nn.quantize.html
index 1fdb1bba8e..b036f11c5f 100644
--- a/docs/build/html/python/_autosummary/mlx.nn.quantize.html
+++ b/docs/build/html/python/_autosummary/mlx.nn.quantize.html
@@ -8,7 +8,7 @@
mlx.nn.quantize — MLX 0.16.0 documentation
+ mlx.nn.quantize — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.nn.value_and_grad.html b/docs/build/html/python/_autosummary/mlx.nn.value_and_grad.html
index 37861975f6..8103ba633f 100644
--- a/docs/build/html/python/_autosummary/mlx.nn.value_and_grad.html
+++ b/docs/build/html/python/_autosummary/mlx.nn.value_and_grad.html
@@ -8,7 +8,7 @@
mlx.nn.value_and_grad — MLX 0.16.0 documentation
+ mlx.nn.value_and_grad — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.optimizers.clip_grad_norm.html b/docs/build/html/python/_autosummary/mlx.optimizers.clip_grad_norm.html
index 705d1603db..8177c05af3 100644
--- a/docs/build/html/python/_autosummary/mlx.optimizers.clip_grad_norm.html
+++ b/docs/build/html/python/_autosummary/mlx.optimizers.clip_grad_norm.html
@@ -8,7 +8,7 @@
mlx.optimizers.clip_grad_norm — MLX 0.16.0 documentation
+ mlx.optimizers.clip_grad_norm — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.utils.tree_flatten.html b/docs/build/html/python/_autosummary/mlx.utils.tree_flatten.html
index 29d0831062..4061036d35 100644
--- a/docs/build/html/python/_autosummary/mlx.utils.tree_flatten.html
+++ b/docs/build/html/python/_autosummary/mlx.utils.tree_flatten.html
@@ -8,7 +8,7 @@
mlx.utils.tree_flatten — MLX 0.16.0 documentation
+ mlx.utils.tree_flatten — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.utils.tree_map.html b/docs/build/html/python/_autosummary/mlx.utils.tree_map.html
index 0475348a41..648d38f571 100644
--- a/docs/build/html/python/_autosummary/mlx.utils.tree_map.html
+++ b/docs/build/html/python/_autosummary/mlx.utils.tree_map.html
@@ -8,7 +8,7 @@
mlx.utils.tree_map — MLX 0.16.0 documentation
+ mlx.utils.tree_map — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
@@ -842,7 +846,7 @@ document.write(`
mlx.utils.tree_map
-tree_map ( fn tree * rest is_leaf = None )
+tree_map ( fn : Callable tree : Any * rest : Tuple [ Any ] is_leaf : Callable | None = None ) → Any Applies fn tree
If rest tree
mlx.utils.tree_map_with_path — MLX 0.16.0 documentation
+ mlx.utils.tree_map_with_path — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
@@ -842,7 +846,7 @@ document.write(`
mlx.utils.tree_map_with_path
-tree_map_with_path ( fn tree * rest is_leaf = None path = None )
+tree_map_with_path ( fn : Callable tree : Any * rest : Tuple [ Any ] is_leaf : Callable | None = None path : Any | None = None ) → Any Applies fn tree
This function is the same tree_map() fn
mlx.utils.tree_reduce — MLX 0.16.0 documentation
+ mlx.utils.tree_reduce — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/mlx.utils.tree_unflatten.html b/docs/build/html/python/_autosummary/mlx.utils.tree_unflatten.html
index 44fc2e2b7c..809cbbf7ca 100644
--- a/docs/build/html/python/_autosummary/mlx.utils.tree_unflatten.html
+++ b/docs/build/html/python/_autosummary/mlx.utils.tree_unflatten.html
@@ -8,7 +8,7 @@
mlx.utils.tree_unflatten — MLX 0.16.0 documentation
+ mlx.utils.tree_unflatten — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/_autosummary/stream_class.html b/docs/build/html/python/_autosummary/stream_class.html
index cf884f50a8..c5ea43dfcb 100644
--- a/docs/build/html/python/_autosummary/stream_class.html
+++ b/docs/build/html/python/_autosummary/stream_class.html
@@ -8,7 +8,7 @@
mlx.core.Stream — MLX 0.16.0 documentation
+ mlx.core.Stream — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/array.html b/docs/build/html/python/array.html
index a7c64af2fc..ea41c62f61 100644
--- a/docs/build/html/python/array.html
+++ b/docs/build/html/python/array.html
@@ -8,7 +8,7 @@
Array — MLX 0.16.0 documentation
+ Array — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/data_types.html b/docs/build/html/python/data_types.html
index 82b28a9c3c..db79fc77f5 100644
--- a/docs/build/html/python/data_types.html
+++ b/docs/build/html/python/data_types.html
@@ -8,7 +8,7 @@
Data Types — MLX 0.16.0 documentation
+ Data Types — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/devices_and_streams.html b/docs/build/html/python/devices_and_streams.html
index b2de85315b..5539dcc179 100644
--- a/docs/build/html/python/devices_and_streams.html
+++ b/docs/build/html/python/devices_and_streams.html
@@ -8,7 +8,7 @@
Devices and Streams — MLX 0.16.0 documentation
+ Devices and Streams — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/distributed.html b/docs/build/html/python/distributed.html
index 5ad8731623..0b3d4c707e 100644
--- a/docs/build/html/python/distributed.html
+++ b/docs/build/html/python/distributed.html
@@ -8,7 +8,7 @@
Distributed Communication — MLX 0.16.0 documentation
+ Distributed Communication — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/fast.html b/docs/build/html/python/fast.html
index bee5cfd9db..799b71d3c5 100644
--- a/docs/build/html/python/fast.html
+++ b/docs/build/html/python/fast.html
@@ -8,7 +8,7 @@
Fast — MLX 0.16.0 documentation
+ Fast — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/fft.html b/docs/build/html/python/fft.html
index 1e75f02cbd..03a1988719 100644
--- a/docs/build/html/python/fft.html
+++ b/docs/build/html/python/fft.html
@@ -8,7 +8,7 @@
FFT — MLX 0.16.0 documentation
+ FFT — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/linalg.html b/docs/build/html/python/linalg.html
index 0574e19536..a41552f2b6 100644
--- a/docs/build/html/python/linalg.html
+++ b/docs/build/html/python/linalg.html
@@ -8,7 +8,7 @@
Linear Algebra — MLX 0.16.0 documentation
+ Linear Algebra — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/metal.html b/docs/build/html/python/metal.html
index 4577ea4f24..4eed55b9ea 100644
--- a/docs/build/html/python/metal.html
+++ b/docs/build/html/python/metal.html
@@ -8,7 +8,7 @@
Metal — MLX 0.16.0 documentation
+ Metal — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/nn.html b/docs/build/html/python/nn.html
index e07acd8f5e..70c2caa76a 100644
--- a/docs/build/html/python/nn.html
+++ b/docs/build/html/python/nn.html
@@ -8,7 +8,7 @@
Neural Networks — MLX 0.16.0 documentation
+ Neural Networks — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
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 35744da587..87b573c442 100644
--- a/docs/build/html/python/nn/_autosummary/mlx.nn.ALiBi.html
+++ b/docs/build/html/python/nn/_autosummary/mlx.nn.ALiBi.html
@@ -8,7 +8,7 @@
mlx.nn.ALiBi — MLX 0.16.0 documentation
+ mlx.nn.ALiBi — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.AvgPool1d.html b/docs/build/html/python/nn/_autosummary/mlx.nn.AvgPool1d.html
index 8313190e5c..9fef328b34 100644
--- a/docs/build/html/python/nn/_autosummary/mlx.nn.AvgPool1d.html
+++ b/docs/build/html/python/nn/_autosummary/mlx.nn.AvgPool1d.html
@@ -8,7 +8,7 @@
mlx.nn.AvgPool1d — MLX 0.16.0 documentation
+ mlx.nn.AvgPool1d — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.AvgPool2d.html b/docs/build/html/python/nn/_autosummary/mlx.nn.AvgPool2d.html
index 166c3d7874..4701a2107b 100644
--- a/docs/build/html/python/nn/_autosummary/mlx.nn.AvgPool2d.html
+++ b/docs/build/html/python/nn/_autosummary/mlx.nn.AvgPool2d.html
@@ -8,7 +8,7 @@
mlx.nn.AvgPool2d — MLX 0.16.0 documentation
+ mlx.nn.AvgPool2d — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.BatchNorm.html b/docs/build/html/python/nn/_autosummary/mlx.nn.BatchNorm.html
index b3e1247f46..3f446ae81e 100644
--- a/docs/build/html/python/nn/_autosummary/mlx.nn.BatchNorm.html
+++ b/docs/build/html/python/nn/_autosummary/mlx.nn.BatchNorm.html
@@ -8,7 +8,7 @@
mlx.nn.BatchNorm — MLX 0.16.0 documentation
+ mlx.nn.BatchNorm — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Conv1d.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Conv1d.html
index aae6a9ab20..c86948f869 100644
--- a/docs/build/html/python/nn/_autosummary/mlx.nn.Conv1d.html
+++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Conv1d.html
@@ -8,7 +8,7 @@
mlx.nn.Conv1d — MLX 0.16.0 documentation
+ mlx.nn.Conv1d — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Conv2d.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Conv2d.html
index 8f2f98e092..4184a5a596 100644
--- a/docs/build/html/python/nn/_autosummary/mlx.nn.Conv2d.html
+++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Conv2d.html
@@ -8,7 +8,7 @@
mlx.nn.Conv2d — MLX 0.16.0 documentation
+ mlx.nn.Conv2d — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Conv3d.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Conv3d.html
index 66847ebf27..2dcf9fb1c1 100644
--- a/docs/build/html/python/nn/_autosummary/mlx.nn.Conv3d.html
+++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Conv3d.html
@@ -8,7 +8,7 @@
mlx.nn.Conv3d — MLX 0.16.0 documentation
+ mlx.nn.Conv3d — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Dropout.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Dropout.html
index 8159179457..cd6263d128 100644
--- a/docs/build/html/python/nn/_autosummary/mlx.nn.Dropout.html
+++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Dropout.html
@@ -8,7 +8,7 @@
mlx.nn.Dropout — MLX 0.16.0 documentation
+ mlx.nn.Dropout — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Dropout2d.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Dropout2d.html
index f9ad1726ac..41f8533718 100644
--- a/docs/build/html/python/nn/_autosummary/mlx.nn.Dropout2d.html
+++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Dropout2d.html
@@ -8,7 +8,7 @@
mlx.nn.Dropout2d — MLX 0.16.0 documentation
+ mlx.nn.Dropout2d — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Dropout3d.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Dropout3d.html
index 039f2fbda3..54b7dc464c 100644
--- a/docs/build/html/python/nn/_autosummary/mlx.nn.Dropout3d.html
+++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Dropout3d.html
@@ -8,7 +8,7 @@
mlx.nn.Dropout3d — MLX 0.16.0 documentation
+ mlx.nn.Dropout3d — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Embedding.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Embedding.html
index e30e287ba3..0016ee76d8 100644
--- a/docs/build/html/python/nn/_autosummary/mlx.nn.Embedding.html
+++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Embedding.html
@@ -8,7 +8,7 @@
mlx.nn.Embedding — MLX 0.16.0 documentation
+ mlx.nn.Embedding — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.GELU.html b/docs/build/html/python/nn/_autosummary/mlx.nn.GELU.html
index c99c6ae3cc..9ad90f14c0 100644
--- a/docs/build/html/python/nn/_autosummary/mlx.nn.GELU.html
+++ b/docs/build/html/python/nn/_autosummary/mlx.nn.GELU.html
@@ -8,7 +8,7 @@
mlx.nn.GELU — MLX 0.16.0 documentation
+ mlx.nn.GELU — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.GLU.html b/docs/build/html/python/nn/_autosummary/mlx.nn.GLU.html
index aa0e250342..731f2a1dec 100644
--- a/docs/build/html/python/nn/_autosummary/mlx.nn.GLU.html
+++ b/docs/build/html/python/nn/_autosummary/mlx.nn.GLU.html
@@ -8,7 +8,7 @@
mlx.nn.GLU — MLX 0.16.0 documentation
+ mlx.nn.GLU — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.GRU.html b/docs/build/html/python/nn/_autosummary/mlx.nn.GRU.html
index 43ea7e5134..752b79a76c 100644
--- a/docs/build/html/python/nn/_autosummary/mlx.nn.GRU.html
+++ b/docs/build/html/python/nn/_autosummary/mlx.nn.GRU.html
@@ -8,7 +8,7 @@
mlx.nn.GRU — MLX 0.16.0 documentation
+ mlx.nn.GRU — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.GroupNorm.html b/docs/build/html/python/nn/_autosummary/mlx.nn.GroupNorm.html
index 1790ace0bb..7482a45ede 100644
--- a/docs/build/html/python/nn/_autosummary/mlx.nn.GroupNorm.html
+++ b/docs/build/html/python/nn/_autosummary/mlx.nn.GroupNorm.html
@@ -8,7 +8,7 @@
mlx.nn.GroupNorm — MLX 0.16.0 documentation
+ mlx.nn.GroupNorm — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.HardShrink.html b/docs/build/html/python/nn/_autosummary/mlx.nn.HardShrink.html
index c8e0676cef..98e5dacca6 100644
--- a/docs/build/html/python/nn/_autosummary/mlx.nn.HardShrink.html
+++ b/docs/build/html/python/nn/_autosummary/mlx.nn.HardShrink.html
@@ -8,7 +8,7 @@
mlx.nn.HardShrink — MLX 0.16.0 documentation
+ mlx.nn.HardShrink — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.HardTanh.html b/docs/build/html/python/nn/_autosummary/mlx.nn.HardTanh.html
index 7f3579f29c..459c534a47 100644
--- a/docs/build/html/python/nn/_autosummary/mlx.nn.HardTanh.html
+++ b/docs/build/html/python/nn/_autosummary/mlx.nn.HardTanh.html
@@ -8,7 +8,7 @@
mlx.nn.HardTanh — MLX 0.16.0 documentation
+ mlx.nn.HardTanh — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Hardswish.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Hardswish.html
index 58faa05004..5e68b69d15 100644
--- a/docs/build/html/python/nn/_autosummary/mlx.nn.Hardswish.html
+++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Hardswish.html
@@ -8,7 +8,7 @@
mlx.nn.Hardswish — MLX 0.16.0 documentation
+ mlx.nn.Hardswish — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.InstanceNorm.html b/docs/build/html/python/nn/_autosummary/mlx.nn.InstanceNorm.html
index e788c69560..a32e8e258a 100644
--- a/docs/build/html/python/nn/_autosummary/mlx.nn.InstanceNorm.html
+++ b/docs/build/html/python/nn/_autosummary/mlx.nn.InstanceNorm.html
@@ -8,7 +8,7 @@
mlx.nn.InstanceNorm — MLX 0.16.0 documentation
+ mlx.nn.InstanceNorm — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.LSTM.html b/docs/build/html/python/nn/_autosummary/mlx.nn.LSTM.html
index dd27fa81da..a534dfeb9e 100644
--- a/docs/build/html/python/nn/_autosummary/mlx.nn.LSTM.html
+++ b/docs/build/html/python/nn/_autosummary/mlx.nn.LSTM.html
@@ -8,7 +8,7 @@
mlx.nn.LSTM — MLX 0.16.0 documentation
+ mlx.nn.LSTM — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.LayerNorm.html b/docs/build/html/python/nn/_autosummary/mlx.nn.LayerNorm.html
index cca02150e0..812a3399dd 100644
--- a/docs/build/html/python/nn/_autosummary/mlx.nn.LayerNorm.html
+++ b/docs/build/html/python/nn/_autosummary/mlx.nn.LayerNorm.html
@@ -8,7 +8,7 @@
mlx.nn.LayerNorm — MLX 0.16.0 documentation
+ mlx.nn.LayerNorm — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.LeakyReLU.html b/docs/build/html/python/nn/_autosummary/mlx.nn.LeakyReLU.html
index 775f7b5d0b..38cee30755 100644
--- a/docs/build/html/python/nn/_autosummary/mlx.nn.LeakyReLU.html
+++ b/docs/build/html/python/nn/_autosummary/mlx.nn.LeakyReLU.html
@@ -8,7 +8,7 @@
mlx.nn.LeakyReLU — MLX 0.16.0 documentation
+ mlx.nn.LeakyReLU — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Linear.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Linear.html
index 2b89db35a5..4dc306c6fd 100644
--- a/docs/build/html/python/nn/_autosummary/mlx.nn.Linear.html
+++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Linear.html
@@ -8,7 +8,7 @@
mlx.nn.Linear — MLX 0.16.0 documentation
+ mlx.nn.Linear — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.MaxPool1d.html b/docs/build/html/python/nn/_autosummary/mlx.nn.MaxPool1d.html
index bde4783ac3..a5380327ea 100644
--- a/docs/build/html/python/nn/_autosummary/mlx.nn.MaxPool1d.html
+++ b/docs/build/html/python/nn/_autosummary/mlx.nn.MaxPool1d.html
@@ -8,7 +8,7 @@
mlx.nn.MaxPool1d — MLX 0.16.0 documentation
+ mlx.nn.MaxPool1d — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.MaxPool2d.html b/docs/build/html/python/nn/_autosummary/mlx.nn.MaxPool2d.html
index 24a1fc1f18..5d752df9e3 100644
--- a/docs/build/html/python/nn/_autosummary/mlx.nn.MaxPool2d.html
+++ b/docs/build/html/python/nn/_autosummary/mlx.nn.MaxPool2d.html
@@ -8,7 +8,7 @@
mlx.nn.MaxPool2d — MLX 0.16.0 documentation
+ mlx.nn.MaxPool2d — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Mish.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Mish.html
index a598c70678..dfde829337 100644
--- a/docs/build/html/python/nn/_autosummary/mlx.nn.Mish.html
+++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Mish.html
@@ -8,7 +8,7 @@
mlx.nn.Mish — MLX 0.16.0 documentation
+ mlx.nn.Mish — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
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 60bed50831..6ffb556a30 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
@@ -8,7 +8,7 @@
mlx.nn.Module.apply — MLX 0.16.0 documentation
+ mlx.nn.Module.apply — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.apply_to_modules.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.apply_to_modules.html
index 4dfa4a5b29..8ece62f0a2 100644
--- a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.apply_to_modules.html
+++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.apply_to_modules.html
@@ -8,7 +8,7 @@
mlx.nn.Module.apply_to_modules — MLX 0.16.0 documentation
+ mlx.nn.Module.apply_to_modules — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.children.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.children.html
index 0e8d369853..9f7fd604bf 100644
--- a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.children.html
+++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.children.html
@@ -8,7 +8,7 @@
mlx.nn.Module.children — MLX 0.16.0 documentation
+ mlx.nn.Module.children — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.eval.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.eval.html
index aabac5df54..363ebaf213 100644
--- a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.eval.html
+++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.eval.html
@@ -8,7 +8,7 @@
mlx.nn.Module.eval — MLX 0.16.0 documentation
+ mlx.nn.Module.eval — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.filter_and_map.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.filter_and_map.html
index 295044fc7b..6448d7c296 100644
--- a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.filter_and_map.html
+++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.filter_and_map.html
@@ -8,7 +8,7 @@
mlx.nn.Module.filter_and_map — MLX 0.16.0 documentation
+ mlx.nn.Module.filter_and_map — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.freeze.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.freeze.html
index f36035cd99..6c3289c371 100644
--- a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.freeze.html
+++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.freeze.html
@@ -8,7 +8,7 @@
mlx.nn.Module.freeze — MLX 0.16.0 documentation
+ mlx.nn.Module.freeze — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.leaf_modules.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.leaf_modules.html
index df2e1a9fa9..b0342998bf 100644
--- a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.leaf_modules.html
+++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.leaf_modules.html
@@ -8,7 +8,7 @@
mlx.nn.Module.leaf_modules — MLX 0.16.0 documentation
+ mlx.nn.Module.leaf_modules — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.load_weights.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.load_weights.html
index 34436d0859..889d7017cf 100644
--- a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.load_weights.html
+++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.load_weights.html
@@ -8,7 +8,7 @@
mlx.nn.Module.load_weights — MLX 0.16.0 documentation
+ mlx.nn.Module.load_weights — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.modules.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.modules.html
index f22c57ceda..3e8c2b0e3d 100644
--- a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.modules.html
+++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.modules.html
@@ -8,7 +8,7 @@
mlx.nn.Module.modules — MLX 0.16.0 documentation
+ mlx.nn.Module.modules — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.named_modules.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.named_modules.html
index df26e9ad85..b6af2babf2 100644
--- a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.named_modules.html
+++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.named_modules.html
@@ -8,7 +8,7 @@
mlx.nn.Module.named_modules — MLX 0.16.0 documentation
+ mlx.nn.Module.named_modules — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.parameters.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.parameters.html
index 67d26fdeef..7c931d5c6a 100644
--- a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.parameters.html
+++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.parameters.html
@@ -8,7 +8,7 @@
mlx.nn.Module.parameters — MLX 0.16.0 documentation
+ mlx.nn.Module.parameters — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.save_weights.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.save_weights.html
index 22a5c226e9..6ba8dc18df 100644
--- a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.save_weights.html
+++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.save_weights.html
@@ -8,7 +8,7 @@
mlx.nn.Module.save_weights — MLX 0.16.0 documentation
+ mlx.nn.Module.save_weights — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.set_dtype.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.set_dtype.html
index 39526fbb74..0da0dfe009 100644
--- a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.set_dtype.html
+++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.set_dtype.html
@@ -8,7 +8,7 @@
mlx.nn.Module.set_dtype — MLX 0.16.0 documentation
+ mlx.nn.Module.set_dtype — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.state.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.state.html
index 7f2777a522..7dbc3dfa28 100644
--- a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.state.html
+++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.state.html
@@ -8,7 +8,7 @@
mlx.nn.Module.state — MLX 0.16.0 documentation
+ mlx.nn.Module.state — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.train.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.train.html
index e16813c95f..aceaac1cfe 100644
--- a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.train.html
+++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.train.html
@@ -8,7 +8,7 @@
mlx.nn.Module.train — MLX 0.16.0 documentation
+ mlx.nn.Module.train — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.trainable_parameters.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.trainable_parameters.html
index 535014516f..714b1c0ad3 100644
--- a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.trainable_parameters.html
+++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.trainable_parameters.html
@@ -8,7 +8,7 @@
mlx.nn.Module.trainable_parameters — MLX 0.16.0 documentation
+ mlx.nn.Module.trainable_parameters — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.training.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.training.html
index d19ae49b30..587bf0e325 100644
--- a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.training.html
+++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.training.html
@@ -8,7 +8,7 @@
mlx.nn.Module.training — MLX 0.16.0 documentation
+ mlx.nn.Module.training — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.unfreeze.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.unfreeze.html
index a54781cf2e..e7eab13a67 100644
--- a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.unfreeze.html
+++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.unfreeze.html
@@ -8,7 +8,7 @@
mlx.nn.Module.unfreeze — MLX 0.16.0 documentation
+ mlx.nn.Module.unfreeze — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.update.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.update.html
index 8a04185597..87faba7f5a 100644
--- a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.update.html
+++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.update.html
@@ -8,7 +8,7 @@
mlx.nn.Module.update — MLX 0.16.0 documentation
+ mlx.nn.Module.update — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.update_modules.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.update_modules.html
index 64e8771225..78952beeab 100644
--- a/docs/build/html/python/nn/_autosummary/mlx.nn.Module.update_modules.html
+++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Module.update_modules.html
@@ -8,7 +8,7 @@
mlx.nn.Module.update_modules — MLX 0.16.0 documentation
+ mlx.nn.Module.update_modules — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.MultiHeadAttention.html b/docs/build/html/python/nn/_autosummary/mlx.nn.MultiHeadAttention.html
index 6c5ef1f219..18a9011897 100644
--- a/docs/build/html/python/nn/_autosummary/mlx.nn.MultiHeadAttention.html
+++ b/docs/build/html/python/nn/_autosummary/mlx.nn.MultiHeadAttention.html
@@ -8,7 +8,7 @@
mlx.nn.MultiHeadAttention — MLX 0.16.0 documentation
+ mlx.nn.MultiHeadAttention — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.PReLU.html b/docs/build/html/python/nn/_autosummary/mlx.nn.PReLU.html
index a0b94ade6f..6e7c2c59b2 100644
--- a/docs/build/html/python/nn/_autosummary/mlx.nn.PReLU.html
+++ b/docs/build/html/python/nn/_autosummary/mlx.nn.PReLU.html
@@ -8,7 +8,7 @@
mlx.nn.PReLU — MLX 0.16.0 documentation
+ mlx.nn.PReLU — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.QuantizedEmbedding.html b/docs/build/html/python/nn/_autosummary/mlx.nn.QuantizedEmbedding.html
index 947b75963e..e46aba950f 100644
--- a/docs/build/html/python/nn/_autosummary/mlx.nn.QuantizedEmbedding.html
+++ b/docs/build/html/python/nn/_autosummary/mlx.nn.QuantizedEmbedding.html
@@ -8,7 +8,7 @@
mlx.nn.QuantizedEmbedding — MLX 0.16.0 documentation
+ mlx.nn.QuantizedEmbedding — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.QuantizedLinear.html b/docs/build/html/python/nn/_autosummary/mlx.nn.QuantizedLinear.html
index 173b4c7d6b..d5f1956e4b 100644
--- a/docs/build/html/python/nn/_autosummary/mlx.nn.QuantizedLinear.html
+++ b/docs/build/html/python/nn/_autosummary/mlx.nn.QuantizedLinear.html
@@ -8,7 +8,7 @@
mlx.nn.QuantizedLinear — MLX 0.16.0 documentation
+ mlx.nn.QuantizedLinear — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.RMSNorm.html b/docs/build/html/python/nn/_autosummary/mlx.nn.RMSNorm.html
index 632c728841..c70e856401 100644
--- a/docs/build/html/python/nn/_autosummary/mlx.nn.RMSNorm.html
+++ b/docs/build/html/python/nn/_autosummary/mlx.nn.RMSNorm.html
@@ -8,7 +8,7 @@
mlx.nn.RMSNorm — MLX 0.16.0 documentation
+ mlx.nn.RMSNorm — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.RNN.html b/docs/build/html/python/nn/_autosummary/mlx.nn.RNN.html
index 2bcd706450..b732708ccc 100644
--- a/docs/build/html/python/nn/_autosummary/mlx.nn.RNN.html
+++ b/docs/build/html/python/nn/_autosummary/mlx.nn.RNN.html
@@ -8,7 +8,7 @@
mlx.nn.RNN — MLX 0.16.0 documentation
+ mlx.nn.RNN — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.ReLU.html b/docs/build/html/python/nn/_autosummary/mlx.nn.ReLU.html
index f54dce6824..80d40d186d 100644
--- a/docs/build/html/python/nn/_autosummary/mlx.nn.ReLU.html
+++ b/docs/build/html/python/nn/_autosummary/mlx.nn.ReLU.html
@@ -8,7 +8,7 @@
mlx.nn.ReLU — MLX 0.16.0 documentation
+ mlx.nn.ReLU — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.ReLU6.html b/docs/build/html/python/nn/_autosummary/mlx.nn.ReLU6.html
index 7664356581..99fd5062dd 100644
--- a/docs/build/html/python/nn/_autosummary/mlx.nn.ReLU6.html
+++ b/docs/build/html/python/nn/_autosummary/mlx.nn.ReLU6.html
@@ -8,7 +8,7 @@
mlx.nn.ReLU6 — MLX 0.16.0 documentation
+ mlx.nn.ReLU6 — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
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 1206f3f228..6d1bd45907 100644
--- a/docs/build/html/python/nn/_autosummary/mlx.nn.RoPE.html
+++ b/docs/build/html/python/nn/_autosummary/mlx.nn.RoPE.html
@@ -8,7 +8,7 @@
mlx.nn.RoPE — MLX 0.16.0 documentation
+ mlx.nn.RoPE — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.SELU.html b/docs/build/html/python/nn/_autosummary/mlx.nn.SELU.html
index 6f091c1515..f06fc7458f 100644
--- a/docs/build/html/python/nn/_autosummary/mlx.nn.SELU.html
+++ b/docs/build/html/python/nn/_autosummary/mlx.nn.SELU.html
@@ -8,7 +8,7 @@
mlx.nn.SELU — MLX 0.16.0 documentation
+ mlx.nn.SELU — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
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 eae7b2e38f..6f81764c36 100644
--- a/docs/build/html/python/nn/_autosummary/mlx.nn.Sequential.html
+++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Sequential.html
@@ -8,7 +8,7 @@
mlx.nn.Sequential — MLX 0.16.0 documentation
+ mlx.nn.Sequential — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.SiLU.html b/docs/build/html/python/nn/_autosummary/mlx.nn.SiLU.html
index 2867372da6..6f3061c97d 100644
--- a/docs/build/html/python/nn/_autosummary/mlx.nn.SiLU.html
+++ b/docs/build/html/python/nn/_autosummary/mlx.nn.SiLU.html
@@ -8,7 +8,7 @@
mlx.nn.SiLU — MLX 0.16.0 documentation
+ mlx.nn.SiLU — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
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 98dfb5b724..78c610cf16 100644
--- a/docs/build/html/python/nn/_autosummary/mlx.nn.SinusoidalPositionalEncoding.html
+++ b/docs/build/html/python/nn/_autosummary/mlx.nn.SinusoidalPositionalEncoding.html
@@ -8,7 +8,7 @@
mlx.nn.SinusoidalPositionalEncoding — MLX 0.16.0 documentation
+ mlx.nn.SinusoidalPositionalEncoding — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Softmax.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Softmax.html
index 0ed4895087..43ee457e03 100644
--- a/docs/build/html/python/nn/_autosummary/mlx.nn.Softmax.html
+++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Softmax.html
@@ -8,7 +8,7 @@
mlx.nn.Softmax — MLX 0.16.0 documentation
+ mlx.nn.Softmax — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Softmin.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Softmin.html
index 8f9e9ee0b8..db214eb14f 100644
--- a/docs/build/html/python/nn/_autosummary/mlx.nn.Softmin.html
+++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Softmin.html
@@ -8,7 +8,7 @@
mlx.nn.Softmin — MLX 0.16.0 documentation
+ mlx.nn.Softmin — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Softplus.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Softplus.html
index 37b9180a3a..c8c81f9893 100644
--- a/docs/build/html/python/nn/_autosummary/mlx.nn.Softplus.html
+++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Softplus.html
@@ -8,7 +8,7 @@
mlx.nn.Softplus — MLX 0.16.0 documentation
+ mlx.nn.Softplus — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Softshrink.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Softshrink.html
index a1a922b2d5..071a7155e8 100644
--- a/docs/build/html/python/nn/_autosummary/mlx.nn.Softshrink.html
+++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Softshrink.html
@@ -8,7 +8,7 @@
mlx.nn.Softshrink — MLX 0.16.0 documentation
+ mlx.nn.Softshrink — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Softsign.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Softsign.html
index c8302a461e..2a2d05d0bb 100644
--- a/docs/build/html/python/nn/_autosummary/mlx.nn.Softsign.html
+++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Softsign.html
@@ -8,7 +8,7 @@
mlx.nn.Softsign — MLX 0.16.0 documentation
+ mlx.nn.Softsign — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Step.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Step.html
index 94933e07c8..4d8ec244c0 100644
--- a/docs/build/html/python/nn/_autosummary/mlx.nn.Step.html
+++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Step.html
@@ -8,7 +8,7 @@
mlx.nn.Step — MLX 0.16.0 documentation
+ mlx.nn.Step — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Tanh.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Tanh.html
index ef8a954e09..b12cfbac6b 100644
--- a/docs/build/html/python/nn/_autosummary/mlx.nn.Tanh.html
+++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Tanh.html
@@ -8,7 +8,7 @@
mlx.nn.Tanh — MLX 0.16.0 documentation
+ mlx.nn.Tanh — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Transformer.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Transformer.html
index e5bbe468f2..14a7f6b394 100644
--- a/docs/build/html/python/nn/_autosummary/mlx.nn.Transformer.html
+++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Transformer.html
@@ -8,7 +8,7 @@
mlx.nn.Transformer — MLX 0.16.0 documentation
+ mlx.nn.Transformer — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.Upsample.html b/docs/build/html/python/nn/_autosummary/mlx.nn.Upsample.html
index 1a15abb022..7af8bdd261 100644
--- a/docs/build/html/python/nn/_autosummary/mlx.nn.Upsample.html
+++ b/docs/build/html/python/nn/_autosummary/mlx.nn.Upsample.html
@@ -8,7 +8,7 @@
mlx.nn.Upsample — MLX 0.16.0 documentation
+ mlx.nn.Upsample — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.init.constant.html b/docs/build/html/python/nn/_autosummary/mlx.nn.init.constant.html
index 51305f4723..63b0a9b603 100644
--- a/docs/build/html/python/nn/_autosummary/mlx.nn.init.constant.html
+++ b/docs/build/html/python/nn/_autosummary/mlx.nn.init.constant.html
@@ -8,7 +8,7 @@
mlx.nn.init.constant — MLX 0.16.0 documentation
+ mlx.nn.init.constant — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.init.glorot_normal.html b/docs/build/html/python/nn/_autosummary/mlx.nn.init.glorot_normal.html
index 7485ea89b3..2d76d4bec6 100644
--- a/docs/build/html/python/nn/_autosummary/mlx.nn.init.glorot_normal.html
+++ b/docs/build/html/python/nn/_autosummary/mlx.nn.init.glorot_normal.html
@@ -8,7 +8,7 @@
mlx.nn.init.glorot_normal — MLX 0.16.0 documentation
+ mlx.nn.init.glorot_normal — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.init.glorot_uniform.html b/docs/build/html/python/nn/_autosummary/mlx.nn.init.glorot_uniform.html
index 24e73687d1..825eecd5ce 100644
--- a/docs/build/html/python/nn/_autosummary/mlx.nn.init.glorot_uniform.html
+++ b/docs/build/html/python/nn/_autosummary/mlx.nn.init.glorot_uniform.html
@@ -8,7 +8,7 @@
mlx.nn.init.glorot_uniform — MLX 0.16.0 documentation
+ mlx.nn.init.glorot_uniform — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.init.he_normal.html b/docs/build/html/python/nn/_autosummary/mlx.nn.init.he_normal.html
index 085c4b3806..f22971cc0d 100644
--- a/docs/build/html/python/nn/_autosummary/mlx.nn.init.he_normal.html
+++ b/docs/build/html/python/nn/_autosummary/mlx.nn.init.he_normal.html
@@ -8,7 +8,7 @@
mlx.nn.init.he_normal — MLX 0.16.0 documentation
+ mlx.nn.init.he_normal — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.init.he_uniform.html b/docs/build/html/python/nn/_autosummary/mlx.nn.init.he_uniform.html
index 199bc1c6c5..8f04b61b73 100644
--- a/docs/build/html/python/nn/_autosummary/mlx.nn.init.he_uniform.html
+++ b/docs/build/html/python/nn/_autosummary/mlx.nn.init.he_uniform.html
@@ -8,7 +8,7 @@
mlx.nn.init.he_uniform — MLX 0.16.0 documentation
+ mlx.nn.init.he_uniform — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.init.identity.html b/docs/build/html/python/nn/_autosummary/mlx.nn.init.identity.html
index ec0766b6ae..258979d76c 100644
--- a/docs/build/html/python/nn/_autosummary/mlx.nn.init.identity.html
+++ b/docs/build/html/python/nn/_autosummary/mlx.nn.init.identity.html
@@ -8,7 +8,7 @@
mlx.nn.init.identity — MLX 0.16.0 documentation
+ mlx.nn.init.identity — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.init.normal.html b/docs/build/html/python/nn/_autosummary/mlx.nn.init.normal.html
index 46b8faec63..ed82d0e6d5 100644
--- a/docs/build/html/python/nn/_autosummary/mlx.nn.init.normal.html
+++ b/docs/build/html/python/nn/_autosummary/mlx.nn.init.normal.html
@@ -8,7 +8,7 @@
mlx.nn.init.normal — MLX 0.16.0 documentation
+ mlx.nn.init.normal — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/nn/_autosummary/mlx.nn.init.uniform.html b/docs/build/html/python/nn/_autosummary/mlx.nn.init.uniform.html
index c3ef9ce4f9..1d6dba783e 100644
--- a/docs/build/html/python/nn/_autosummary/mlx.nn.init.uniform.html
+++ b/docs/build/html/python/nn/_autosummary/mlx.nn.init.uniform.html
@@ -8,7 +8,7 @@
mlx.nn.init.uniform — MLX 0.16.0 documentation
+ mlx.nn.init.uniform — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.elu.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.elu.html
index c4606219e0..f8392f6f58 100644
--- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.elu.html
+++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.elu.html
@@ -8,7 +8,7 @@
mlx.nn.elu — MLX 0.16.0 documentation
+ mlx.nn.elu — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.gelu.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.gelu.html
index 538c907a60..d98a9b4deb 100644
--- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.gelu.html
+++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.gelu.html
@@ -8,7 +8,7 @@
mlx.nn.gelu — MLX 0.16.0 documentation
+ mlx.nn.gelu — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.gelu_approx.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.gelu_approx.html
index c30e0215c4..345a0e4052 100644
--- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.gelu_approx.html
+++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.gelu_approx.html
@@ -8,7 +8,7 @@
mlx.nn.gelu_approx — MLX 0.16.0 documentation
+ mlx.nn.gelu_approx — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.gelu_fast_approx.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.gelu_fast_approx.html
index 227bff6c20..771f9daef3 100644
--- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.gelu_fast_approx.html
+++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.gelu_fast_approx.html
@@ -8,7 +8,7 @@
mlx.nn.gelu_fast_approx — MLX 0.16.0 documentation
+ mlx.nn.gelu_fast_approx — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
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
index 54656f843d..992d1cc101 100644
--- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.glu.html
+++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.glu.html
@@ -8,7 +8,7 @@
mlx.nn.glu — MLX 0.16.0 documentation
+ mlx.nn.glu — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.hard_shrink.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.hard_shrink.html
index b24fa24b3e..5a6478b2b4 100644
--- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.hard_shrink.html
+++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.hard_shrink.html
@@ -8,7 +8,7 @@
mlx.nn.hard_shrink — MLX 0.16.0 documentation
+ mlx.nn.hard_shrink — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.hard_tanh.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.hard_tanh.html
index de6f14e02c..f1d32e935a 100644
--- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.hard_tanh.html
+++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.hard_tanh.html
@@ -8,7 +8,7 @@
mlx.nn.hard_tanh — MLX 0.16.0 documentation
+ mlx.nn.hard_tanh — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.hardswish.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.hardswish.html
index 62bea8396b..2082622cc2 100644
--- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.hardswish.html
+++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.hardswish.html
@@ -8,7 +8,7 @@
mlx.nn.hardswish — MLX 0.16.0 documentation
+ mlx.nn.hardswish — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.leaky_relu.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.leaky_relu.html
index b5fee83b74..f10ca6d119 100644
--- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.leaky_relu.html
+++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.leaky_relu.html
@@ -8,7 +8,7 @@
mlx.nn.leaky_relu — MLX 0.16.0 documentation
+ mlx.nn.leaky_relu — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.log_sigmoid.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.log_sigmoid.html
index 9c2762f0c3..b10b2e2677 100644
--- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.log_sigmoid.html
+++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.log_sigmoid.html
@@ -8,7 +8,7 @@
mlx.nn.log_sigmoid — MLX 0.16.0 documentation
+ mlx.nn.log_sigmoid — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.log_softmax.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.log_softmax.html
index 7824383927..e8fa10d084 100644
--- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.log_softmax.html
+++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.log_softmax.html
@@ -8,7 +8,7 @@
mlx.nn.log_softmax — MLX 0.16.0 documentation
+ mlx.nn.log_softmax — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.binary_cross_entropy.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.binary_cross_entropy.html
index 12ba32099f..c8844ece70 100644
--- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.binary_cross_entropy.html
+++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.binary_cross_entropy.html
@@ -8,7 +8,7 @@
mlx.nn.losses.binary_cross_entropy — MLX 0.16.0 documentation
+ mlx.nn.losses.binary_cross_entropy — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
@@ -844,6 +848,10 @@ document.write(`
class binary_cross_entropy ( inputs : array targets : array weights : array | None = None with_logits : bool = True reduction : Literal [ 'none' , 'mean' , 'sum' ] = 'mean' )
Computes the binary cross entropy loss.
+By default, this function takes the pre-sigmoid logits, which results in a faster
+and more precise loss. For improved numerical stability when with_logits=False -100
Parameters:
diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.cosine_similarity_loss.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.cosine_similarity_loss.html
index e204d7a1ae..0500a996dc 100644
--- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.cosine_similarity_loss.html
+++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.cosine_similarity_loss.html
@@ -8,7 +8,7 @@
mlx.nn.losses.cosine_similarity_loss — MLX 0.16.0 documentation
+ mlx.nn.losses.cosine_similarity_loss — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.cross_entropy.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.cross_entropy.html
index 70223c02df..ee92ba8582 100644
--- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.cross_entropy.html
+++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.cross_entropy.html
@@ -8,7 +8,7 @@
mlx.nn.losses.cross_entropy — MLX 0.16.0 documentation
+ mlx.nn.losses.cross_entropy — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.gaussian_nll_loss.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.gaussian_nll_loss.html
index cd46682d41..21c2255e45 100644
--- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.gaussian_nll_loss.html
+++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.gaussian_nll_loss.html
@@ -8,7 +8,7 @@
mlx.nn.losses.gaussian_nll_loss — MLX 0.16.0 documentation
+ mlx.nn.losses.gaussian_nll_loss — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.hinge_loss.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.hinge_loss.html
index 826e478d31..3a1e83d798 100644
--- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.hinge_loss.html
+++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.hinge_loss.html
@@ -8,7 +8,7 @@
mlx.nn.losses.hinge_loss — MLX 0.16.0 documentation
+ mlx.nn.losses.hinge_loss — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.huber_loss.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.huber_loss.html
index 22bc6dae2b..5aea944b78 100644
--- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.huber_loss.html
+++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.huber_loss.html
@@ -8,7 +8,7 @@
mlx.nn.losses.huber_loss — MLX 0.16.0 documentation
+ mlx.nn.losses.huber_loss — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.kl_div_loss.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.kl_div_loss.html
index 8d3e60ed39..c377bf25c5 100644
--- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.kl_div_loss.html
+++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.kl_div_loss.html
@@ -8,7 +8,7 @@
mlx.nn.losses.kl_div_loss — MLX 0.16.0 documentation
+ mlx.nn.losses.kl_div_loss — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.l1_loss.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.l1_loss.html
index 3b6183c43e..f500ba8d0f 100644
--- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.l1_loss.html
+++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.l1_loss.html
@@ -8,7 +8,7 @@
mlx.nn.losses.l1_loss — MLX 0.16.0 documentation
+ mlx.nn.losses.l1_loss — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.log_cosh_loss.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.log_cosh_loss.html
index 6c44166771..43061455fc 100644
--- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.log_cosh_loss.html
+++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.log_cosh_loss.html
@@ -8,7 +8,7 @@
mlx.nn.losses.log_cosh_loss — MLX 0.16.0 documentation
+ mlx.nn.losses.log_cosh_loss — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.margin_ranking_loss.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.margin_ranking_loss.html
index b961a5d2b1..7c9e6ce48f 100644
--- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.margin_ranking_loss.html
+++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.margin_ranking_loss.html
@@ -8,7 +8,7 @@
mlx.nn.losses.margin_ranking_loss — MLX 0.16.0 documentation
+ mlx.nn.losses.margin_ranking_loss — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.mse_loss.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.mse_loss.html
index f70d963d18..8a7af2f754 100644
--- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.mse_loss.html
+++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.mse_loss.html
@@ -8,7 +8,7 @@
mlx.nn.losses.mse_loss — MLX 0.16.0 documentation
+ mlx.nn.losses.mse_loss — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.nll_loss.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.nll_loss.html
index 794c103987..435976e5c8 100644
--- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.nll_loss.html
+++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.nll_loss.html
@@ -8,7 +8,7 @@
mlx.nn.losses.nll_loss — MLX 0.16.0 documentation
+ mlx.nn.losses.nll_loss — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.smooth_l1_loss.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.smooth_l1_loss.html
index 560e13de0c..76093ff234 100644
--- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.smooth_l1_loss.html
+++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.smooth_l1_loss.html
@@ -8,7 +8,7 @@
mlx.nn.losses.smooth_l1_loss — MLX 0.16.0 documentation
+ mlx.nn.losses.smooth_l1_loss — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.triplet_loss.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.triplet_loss.html
index e2ae6be03c..f5a8c26ce7 100644
--- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.triplet_loss.html
+++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.losses.triplet_loss.html
@@ -8,7 +8,7 @@
mlx.nn.losses.triplet_loss — MLX 0.16.0 documentation
+ mlx.nn.losses.triplet_loss — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.mish.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.mish.html
index 9a9a62b918..e71fdd2944 100644
--- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.mish.html
+++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.mish.html
@@ -8,7 +8,7 @@
mlx.nn.mish — MLX 0.16.0 documentation
+ mlx.nn.mish — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.prelu.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.prelu.html
index eaf286b49b..f1f6f0785d 100644
--- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.prelu.html
+++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.prelu.html
@@ -8,7 +8,7 @@
mlx.nn.prelu — MLX 0.16.0 documentation
+ mlx.nn.prelu — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.relu.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.relu.html
index 1b7b48ff63..7918dc7222 100644
--- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.relu.html
+++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.relu.html
@@ -8,7 +8,7 @@
mlx.nn.relu — MLX 0.16.0 documentation
+ mlx.nn.relu — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
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
index 0808c23ef7..760b04d3a8 100644
--- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.relu6.html
+++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.relu6.html
@@ -8,7 +8,7 @@
mlx.nn.relu6 — MLX 0.16.0 documentation
+ mlx.nn.relu6 — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.selu.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.selu.html
index f3b43fce22..fd702a7be6 100644
--- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.selu.html
+++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.selu.html
@@ -8,7 +8,7 @@
mlx.nn.selu — MLX 0.16.0 documentation
+ mlx.nn.selu — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
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
index 167802cc27..845f214883 100644
--- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.sigmoid.html
+++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.sigmoid.html
@@ -8,7 +8,7 @@
mlx.nn.sigmoid — MLX 0.16.0 documentation
+ mlx.nn.sigmoid — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
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 d05f4c995e..aa65e46a15 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
@@ -8,7 +8,7 @@
mlx.nn.silu — MLX 0.16.0 documentation
+ mlx.nn.silu — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
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
index 0006e0decf..fe0e6fa1d2 100644
--- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.softmax.html
+++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.softmax.html
@@ -8,7 +8,7 @@
mlx.nn.softmax — MLX 0.16.0 documentation
+ mlx.nn.softmax — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.softmin.html b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.softmin.html
index e9713bc713..e8f2fd38ec 100644
--- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.softmin.html
+++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.softmin.html
@@ -8,7 +8,7 @@
mlx.nn.softmin — MLX 0.16.0 documentation
+ mlx.nn.softmin — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
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
index 639268b6e6..ac4c9990ae 100644
--- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.softplus.html
+++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.softplus.html
@@ -8,7 +8,7 @@
mlx.nn.softplus — MLX 0.16.0 documentation
+ mlx.nn.softplus — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
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 001ec10cdd..99cfef2e3a 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
@@ -8,7 +8,7 @@
mlx.nn.softshrink — MLX 0.16.0 documentation
+ mlx.nn.softshrink — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
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 e03640c3ce..a0ef6a7f42 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
@@ -8,7 +8,7 @@
mlx.nn.step — MLX 0.16.0 documentation
+ mlx.nn.step — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
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
index 4aaeaa6a0e..d47ac4b047 100644
--- a/docs/build/html/python/nn/_autosummary_functions/mlx.nn.tanh.html
+++ b/docs/build/html/python/nn/_autosummary_functions/mlx.nn.tanh.html
@@ -8,7 +8,7 @@
mlx.nn.tanh — MLX 0.16.0 documentation
+ mlx.nn.tanh — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/nn/functions.html b/docs/build/html/python/nn/functions.html
index cf1b1507d4..11573a5d62 100644
--- a/docs/build/html/python/nn/functions.html
+++ b/docs/build/html/python/nn/functions.html
@@ -8,7 +8,7 @@
Functions — MLX 0.16.0 documentation
+ Functions — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/nn/init.html b/docs/build/html/python/nn/init.html
index 9acc3c8d41..f65c6caeba 100644
--- a/docs/build/html/python/nn/init.html
+++ b/docs/build/html/python/nn/init.html
@@ -8,7 +8,7 @@
Initializers — MLX 0.16.0 documentation
+ Initializers — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/nn/layers.html b/docs/build/html/python/nn/layers.html
index b0eb5bb29c..1c08a6569d 100644
--- a/docs/build/html/python/nn/layers.html
+++ b/docs/build/html/python/nn/layers.html
@@ -8,7 +8,7 @@
Layers — MLX 0.16.0 documentation
+ Layers — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/nn/losses.html b/docs/build/html/python/nn/losses.html
index 6537707405..e5072b9f7c 100644
--- a/docs/build/html/python/nn/losses.html
+++ b/docs/build/html/python/nn/losses.html
@@ -8,7 +8,7 @@
Loss Functions — MLX 0.16.0 documentation
+ Loss Functions — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/nn/module.html b/docs/build/html/python/nn/module.html
index fc4be847cd..87527034de 100644
--- a/docs/build/html/python/nn/module.html
+++ b/docs/build/html/python/nn/module.html
@@ -8,7 +8,7 @@
Module — MLX 0.16.0 documentation
+ Module — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/ops.html b/docs/build/html/python/ops.html
index f0daede6d5..546c3b8aaa 100644
--- a/docs/build/html/python/ops.html
+++ b/docs/build/html/python/ops.html
@@ -8,7 +8,7 @@
Operations — MLX 0.16.0 documentation
+ Operations — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
@@ -979,6 +983,12 @@ document.write(`
divmod
Element-wise quotient and remainder.
einsum
Perform the Einstein summation convention on the operands.
einsum_path
Compute the contraction order for the given Einstein summation.
equal
Element-wise equality.
greater_equal
Element-wise greater or equal.
hadamard_transform
hadamard_transform
Perform the Walsh-Hadamard transform along the final axis.
identity
multiply
Element-wise multiplication.
negative
nan_to_num
Replace NaN and Inf values with finite numbers.
negative
Element-wise negation.
not_equal
not_equal
Element-wise not equal.
ones
ones
Construct an array of ones.
ones_like
ones_like
An array of ones like the input.
outer
outer
Compute the outer product of two 1-D arrays, if the array's passed are not 1-D a flatten op will be run beforehand.
partition
partition
Returns a partitioned copy of the array such that the smaller kth
pad
pad
Pad an array with a constant value
power
power
Element-wise power operation.
prod
prod
An product reduction over the given axes.
quantize
quantize
Quantize the matrix w bits
quantized_matmul
quantized_matmul
Perform the matrix multiplication with the quantized matrix w
radians
radians
Convert angles from degrees to radians.
reciprocal
reciprocal
Element-wise reciprocal.
remainder
remainder
Element-wise remainder of division.
repeat
repeat
Repeat an array along a specified axis.
reshape
reshape
Reshape an array while preserving the size.
right_shift
right_shift
Element-wise right shift.
round
round
Round to the given number of decimals.
rsqrt
rsqrt
Element-wise reciprocal and square root.
save
save
Save the array to a binary file in .npy
savez
savez
Save several arrays to a binary file in uncompressed .npz
savez_compressed
savez_compressed
Save several arrays to a binary file in compressed .npz
save_gguf
save_gguf
Save array(s) to a binary file in .gguf
save_safetensors
save_safetensors
Save array(s) to a binary file in .safetensors
sigmoid
sigmoid
Element-wise logistic sigmoid.
sign
sign
Element-wise sign.
sin
sin
Element-wise sine.
sinh
sinh
Element-wise hyperbolic sine.
softmax
softmax
Perform the softmax along the given axis.
sort
sort
Returns a sorted copy of the array.
split
split
Split an array along a given axis.
sqrt
sqrt
Element-wise square root.
square
square
Element-wise square.
squeeze
squeeze
Remove length one axes from an array.
stack
stack
Stacks the arrays along a new axis.
std
std
Compute the standard deviation(s) over the given axes.
stop_gradient
stop_gradient
Stop gradients from being computed.
subtract
subtract
Element-wise subtraction.
sum
sum
Sum reduce the array over the given axes.
swapaxes
swapaxes
Swap two axes of an array.
take
take
Take elements along an axis.
take_along_axis
take_along_axis
Take values along an axis at the specified indices.
tan
tan
Element-wise tangent.
tanh
tanh
Element-wise hyperbolic tangent.
tensordot
tensordot
Compute the tensor dot product along the specified axes.
tile
tile
Construct an array by repeating a reps
topk
topk
Returns the k
trace
trace
Return the sum along a specified diagonal in the given array.
transpose
transpose
Transpose the dimensions of the array.
tri
tri
An array with ones at and below the given diagonal and zeros elsewhere.
tril
tril
Zeros the array above the given diagonal.
triu
triu
Zeros the array below the given diagonal.
var
var
Compute the variance(s) over the given axes.
view
view
View the array as a different type.
where
where
Select from x y condition
zeros
zeros
Construct an array of zeros.
zeros_like
zeros_like
An array of zeros like the input.
Optimizers — MLX 0.16.0 documentation
+ Optimizers — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
@@ -803,7 +807,9 @@ document.write(`
`);
-
+
@@ -819,6 +825,16 @@ document.write(`
@@ -854,6 +870,37 @@ model’s parameters and the optimizer state .
mx . eval ( model . parameters (), optimizer . state )
+
+Saving and Loading
+To serialize an optimizer, save its state. To load an optimizer, load and set
+the saved state. Here’s a simple example:
+import mlx.core as mx
+from mlx.utils import tree_flatten , tree_unflatten
+import mlx.optimizers as optim
+
+optimizer = optim . Adam ( learning_rate = 1e-2 )
+
+# Perform some updates with the optimizer
+model = { "w" : mx . zeros (( 5 , 5 ))}
+grads = { "w" : mx . ones (( 5 , 5 ))}
+optimizer . update ( model , grads )
+
+# Save the state
+state = tree_flatten ( optimizer . state )
+mx . save_safetensors ( "optimizer.safetensors" , dict ( state ))
+
+# Later on, for example when loading from a checkpoint,
+# recreate the optimizer and load the state
+optimizer = optim . Adam ( learning_rate = 1e-2 )
+
+state = tree_unflatten ( list ( mx . load ( "optimizer.safetensors" ) . items ()))
+optimizer . state = state
+
+
Note, not every optimizer configuation parameter is saved in the state. For
+example, for Adam the learning rate is saved but the betas eps
Optimizer
@@ -947,6 +994,7 @@ model’s parameters and the optimizer state .
+
@@ -985,6 +1033,22 @@ model’s parameters and the optimizer state .
+
+
Transforms
diff --git a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Adafactor.html b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Adafactor.html
index d2420a300e..9c1569a586 100644
--- a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Adafactor.html
+++ b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Adafactor.html
@@ -8,7 +8,7 @@
mlx.optimizers.Adafactor — MLX 0.16.0 documentation
+ mlx.optimizers.Adafactor — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Adagrad.html b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Adagrad.html
index 9a5d2ceb30..9e55aa2240 100644
--- a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Adagrad.html
+++ b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Adagrad.html
@@ -8,7 +8,7 @@
mlx.optimizers.Adagrad — MLX 0.16.0 documentation
+ mlx.optimizers.Adagrad — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Adam.html b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Adam.html
index 3ab860bbb9..ef6e72ac8b 100644
--- a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Adam.html
+++ b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Adam.html
@@ -8,7 +8,7 @@
mlx.optimizers.Adam — MLX 0.16.0 documentation
+ mlx.optimizers.Adam — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.AdamW.html b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.AdamW.html
index 9b34ea83fc..e305f9f31a 100644
--- a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.AdamW.html
+++ b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.AdamW.html
@@ -8,7 +8,7 @@
mlx.optimizers.AdamW — MLX 0.16.0 documentation
+ mlx.optimizers.AdamW — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Adamax.html b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Adamax.html
index 6e75250ff8..3402bf78aa 100644
--- a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Adamax.html
+++ b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Adamax.html
@@ -8,7 +8,7 @@
mlx.optimizers.Adamax — MLX 0.16.0 documentation
+ mlx.optimizers.Adamax — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Lion.html b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Lion.html
index 5d185d1cce..834bb80bd3 100644
--- a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Lion.html
+++ b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Lion.html
@@ -8,7 +8,7 @@
mlx.optimizers.Lion — MLX 0.16.0 documentation
+ mlx.optimizers.Lion — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Optimizer.apply_gradients.html b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Optimizer.apply_gradients.html
index ac978b275b..295e9788ab 100644
--- a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Optimizer.apply_gradients.html
+++ b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Optimizer.apply_gradients.html
@@ -8,7 +8,7 @@
mlx.optimizers.Optimizer.apply_gradients — MLX 0.16.0 documentation
+ mlx.optimizers.Optimizer.apply_gradients — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Optimizer.init.html b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Optimizer.init.html
index ac0add56f2..6bd5e76bb8 100644
--- a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Optimizer.init.html
+++ b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Optimizer.init.html
@@ -8,7 +8,7 @@
mlx.optimizers.Optimizer.init — MLX 0.16.0 documentation
+ mlx.optimizers.Optimizer.init — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Optimizer.state.html b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Optimizer.state.html
index 4e82436ab7..d59b3227a9 100644
--- a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Optimizer.state.html
+++ b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Optimizer.state.html
@@ -8,7 +8,7 @@
mlx.optimizers.Optimizer.state — MLX 0.16.0 documentation
+ mlx.optimizers.Optimizer.state — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Optimizer.update.html b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Optimizer.update.html
index 2e0590ae9c..01a13c1d32 100644
--- a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Optimizer.update.html
+++ b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.Optimizer.update.html
@@ -8,7 +8,7 @@
mlx.optimizers.Optimizer.update — MLX 0.16.0 documentation
+ mlx.optimizers.Optimizer.update — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.RMSprop.html b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.RMSprop.html
index d07f36764a..4acd664a35 100644
--- a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.RMSprop.html
+++ b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.RMSprop.html
@@ -8,7 +8,7 @@
mlx.optimizers.RMSprop — MLX 0.16.0 documentation
+ mlx.optimizers.RMSprop — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.SGD.html b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.SGD.html
index faf853f96a..ed5854ec54 100644
--- a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.SGD.html
+++ b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.SGD.html
@@ -8,7 +8,7 @@
mlx.optimizers.SGD — MLX 0.16.0 documentation
+ mlx.optimizers.SGD — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.cosine_decay.html b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.cosine_decay.html
index 65ae4f2e00..7f60ac15f2 100644
--- a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.cosine_decay.html
+++ b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.cosine_decay.html
@@ -8,7 +8,7 @@
mlx.optimizers.cosine_decay — MLX 0.16.0 documentation
+ mlx.optimizers.cosine_decay — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
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 e47c617dd6..1bebc60c37 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
@@ -8,7 +8,7 @@
mlx.optimizers.exponential_decay — MLX 0.16.0 documentation
+ mlx.optimizers.exponential_decay — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
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
index bf1bb29de6..c5ef73578d 100644
--- a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.join_schedules.html
+++ b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.join_schedules.html
@@ -8,7 +8,7 @@
mlx.optimizers.join_schedules — MLX 0.16.0 documentation
+ mlx.optimizers.join_schedules — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
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
index f9c5e0608b..8b9ee6ebde 100644
--- a/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.linear_schedule.html
+++ b/docs/build/html/python/optimizers/_autosummary/mlx.optimizers.linear_schedule.html
@@ -8,7 +8,7 @@
mlx.optimizers.linear_schedule — MLX 0.16.0 documentation
+ mlx.optimizers.linear_schedule — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
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 a891513be9..5ab5faaeb4 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
@@ -8,7 +8,7 @@
mlx.optimizers.step_decay — MLX 0.16.0 documentation
+ mlx.optimizers.step_decay — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/optimizers/common_optimizers.html b/docs/build/html/python/optimizers/common_optimizers.html
index 6e596cdb24..10418ec70b 100644
--- a/docs/build/html/python/optimizers/common_optimizers.html
+++ b/docs/build/html/python/optimizers/common_optimizers.html
@@ -8,7 +8,7 @@
Common Optimizers — MLX 0.16.0 documentation
+ Common Optimizers — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/optimizers/optimizer.html b/docs/build/html/python/optimizers/optimizer.html
index ac02737c2d..56fea43f2b 100644
--- a/docs/build/html/python/optimizers/optimizer.html
+++ b/docs/build/html/python/optimizers/optimizer.html
@@ -8,7 +8,7 @@
Optimizer — MLX 0.16.0 documentation
+ Optimizer — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/optimizers/schedulers.html b/docs/build/html/python/optimizers/schedulers.html
index 84025260e9..a9cb0de38d 100644
--- a/docs/build/html/python/optimizers/schedulers.html
+++ b/docs/build/html/python/optimizers/schedulers.html
@@ -8,7 +8,7 @@
Schedulers — MLX 0.16.0 documentation
+ Schedulers — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/python/random.html b/docs/build/html/python/random.html
index 159ac43a2a..90e93be58e 100644
--- a/docs/build/html/python/random.html
+++ b/docs/build/html/python/random.html
@@ -8,7 +8,7 @@
Random — MLX 0.16.0 documentation
+ Random — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
@@ -883,6 +887,9 @@ we use a splittable version of Threefry, which is a counter-based PRNG.
uniform
Generate uniformly distributed random numbers.
laplace
Sample numbers from a Laplace distribution.
diff --git a/docs/build/html/python/transforms.html b/docs/build/html/python/transforms.html
index 8ac2f33e09..44516b1cb9 100644
--- a/docs/build/html/python/transforms.html
+++ b/docs/build/html/python/transforms.html
@@ -8,7 +8,7 @@
Transforms — MLX 0.16.0 documentation
+ Transforms — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms Transforms
template<typename val_t , typename idx_t , bool ARG_SORT, short BLOCK_THREADS, short N_PER_THREAD> void mb_block_sort (const device val_t *inp, device val_t *out_vals, device idx_t *out_idxs, const constant int &size_sorted_axis, const constant int &stride_sorted_axis, const constant int &nc_dim, const device int *nc_shape, const device size_t *nc_strides, uint3 tid, uint3 lid) template<typename val_t , typename idx_t , bool ARG_SORT, short BLOCK_THREADS, short N_PER_THREAD> void mb_block_partition (device idx_t *block_partitions, const device val_t *dev_vals, const device idx_t *dev_idxs, const constant int &size_sorted_axis, const constant int &merge_tiles, uint3 tid, uint3 lid, uint3 tgp_dims) template<typename val_t , typename idx_t , bool ARG_SORT, short BLOCK_THREADS, short N_PER_THREAD> void mb_block_partition (device idx_t *block_partitions, const device val_t *dev_vals, const device idx_t *dev_idxs, const constant int &size_sorted_axis, const constant int &merge_tiles, const constant int &n_blocks, uint3 tid, uint3 lid, uint3 tgp_dims) template<typename val_t , typename idx_t , bool ARG_SORT, short BLOCK_THREADS, short N_PER_THREAD, typename CompareOp = LessThan<val_t>> void mb_block_merge (const device idx_t *block_partitions, const device val_t *dev_vals_in, const device idx_t *dev_idxs_in, device val_t *dev_vals_out, device idx_t *dev_idxs_out, const constant int &size_sorted_axis, const constant int &merge_tiles, const constant int &num_tiles, uint3 tid, uint3 lid) ◆ ◆
@@ -384,6 +384,11 @@ template<typename val_t , typename idx_t , bool ARG_SORT, short BLOCK_THREADS
const constant int & merge_tiles ,
+
+ const constant int & n_blocks ,
525 [[kernel, max_total_threads_per_threadgroup(BLOCK_THREADS)]]
void
-
-
527 device idx_t* block_partitions [[buffer(0)]],
-
528 const device val_t* dev_vals [[buffer(1)]],
-
529 const device idx_t* dev_idxs [[buffer(2)]],
-
530 const constant
int & size_sorted_axis [[buffer(3)]],
-
531 const constant
int & merge_tiles [[buffer(4)]],
+
+
+
526 device idx_t* block_partitions [[buffer(0)]],
+
527 const device val_t* dev_vals [[buffer(1)]],
+
528 const device idx_t* dev_idxs [[buffer(2)]],
+
529 const constant
int & size_sorted_axis [[buffer(3)]],
+
530 const constant
int & merge_tiles [[buffer(4)]],
+
531 const constant
int & n_blocks [[buffer(5)]],
532 uint3 tid [[threadgroup_position_in_grid]],
533 uint3 lid [[thread_position_in_threadgroup]],
534 uint3 tgp_dims [[threads_per_threadgroup]]) {
@@ -657,152 +657,158 @@ $(function() { codefold.init(0); });
543 dev_vals += tid.y * size_sorted_axis;
544 dev_idxs += tid.y * size_sorted_axis;
-
-
547 int merge_group = lid.x / merge_tiles;
-
548 int merge_lane = lid.x % merge_tiles;
-
-
550 int sort_sz = sort_kernel::N_PER_BLOCK * merge_tiles;
-
551 int sort_st = sort_kernel::N_PER_BLOCK * merge_tiles * merge_group;
-
-
553 int A_st =
min (size_sorted_axis, sort_st);
-
554 int A_ed =
min (size_sorted_axis, sort_st + sort_sz / 2);
-
-
556 int B_ed =
min (size_sorted_axis, B_st + sort_sz / 2);
-
-
558 int partition_at =
min (B_ed - A_st, sort_kernel::N_PER_BLOCK * merge_lane);
-
559 int partition = sort_kernel::merge_partition(
-
560 dev_vals + A_st, dev_vals + B_st, A_ed - A_st, B_ed - B_st, partition_at);
-
-
562 block_partitions[lid.x] = A_st + partition;
-
+
546 for (
int i = lid.x; i <= n_blocks; i += tgp_dims.x) {
+
+
548 int merge_group = i / merge_tiles;
+
549 int merge_lane = i % merge_tiles;
+
+
551 int sort_sz = sort_kernel::N_PER_BLOCK * merge_tiles;
+
552 int sort_st = sort_kernel::N_PER_BLOCK * merge_tiles * merge_group;
+
+
554 int A_st =
min (size_sorted_axis, sort_st);
+
555 int A_ed =
min (size_sorted_axis, sort_st + sort_sz / 2);
+
+
557 int B_ed =
min (size_sorted_axis, B_st + sort_sz / 2);
+
+
559 int partition_at =
min (B_ed - A_st, sort_kernel::N_PER_BLOCK * merge_lane);
+
560 int partition = sort_kernel::merge_partition(
+
+
+
+
+
+
+
567 block_partitions[i] = A_st + partition;
+
+
-
-
-
-
-
-
-
-
-
572 [[kernel, max_total_threads_per_threadgroup(BLOCK_THREADS)]]
void
-
-
-
574 const device idx_t* block_partitions [[buffer(0)]],
-
575 const device val_t* dev_vals_in [[buffer(1)]],
-
576 const device idx_t* dev_idxs_in [[buffer(2)]],
-
577 device val_t* dev_vals_out [[buffer(3)]],
-
578 device idx_t* dev_idxs_out [[buffer(4)]],
-
579 const constant
int & size_sorted_axis [[buffer(5)]],
-
580 const constant
int & merge_tiles [[buffer(6)]],
-
581 const constant
int & num_tiles [[buffer(7)]],
-
582 uint3 tid [[threadgroup_position_in_grid]],
-
583 uint3 lid [[thread_position_in_threadgroup]]) {
-
-
-
-
-
-
-
-
-
592 using block_sort_t =
typename sort_kernel::block_merge_sort_t;
-
-
594 block_partitions += tid.y * (num_tiles + 1);
-
595 dev_vals_in += tid.y * size_sorted_axis;
-
596 dev_idxs_in += tid.y * size_sorted_axis;
-
597 dev_vals_out += tid.y * size_sorted_axis;
-
598 dev_idxs_out += tid.y * size_sorted_axis;
+
+
+
+
+
+
+
+
+
578 [[kernel, max_total_threads_per_threadgroup(BLOCK_THREADS)]]
void
+
+
+
580 const device idx_t* block_partitions [[buffer(0)]],
+
581 const device val_t* dev_vals_in [[buffer(1)]],
+
582 const device idx_t* dev_idxs_in [[buffer(2)]],
+
583 device val_t* dev_vals_out [[buffer(3)]],
+
584 device idx_t* dev_idxs_out [[buffer(4)]],
+
585 const constant
int & size_sorted_axis [[buffer(5)]],
+
586 const constant
int & merge_tiles [[buffer(6)]],
+
587 const constant
int & num_tiles [[buffer(7)]],
+
588 uint3 tid [[threadgroup_position_in_grid]],
+
589 uint3 lid [[thread_position_in_threadgroup]]) {
+
+
+
+
+
+
+
+
+
598 using block_sort_t =
typename sort_kernel::block_merge_sort_t;
-
600 int block_idx = tid.x;
-
601 int merge_group = block_idx / merge_tiles;
-
602 int sort_st = sort_kernel::N_PER_BLOCK * merge_tiles * merge_group;
-
603 int sort_sz = sort_kernel::N_PER_BLOCK * merge_tiles;
-
604 int sort_md = sort_kernel::N_PER_BLOCK * block_idx - sort_st;
+
600 block_partitions += tid.y * (num_tiles + 1);
+
601 dev_vals_in += tid.y * size_sorted_axis;
+
602 dev_idxs_in += tid.y * size_sorted_axis;
+
603 dev_vals_out += tid.y * size_sorted_axis;
+
604 dev_idxs_out += tid.y * size_sorted_axis;
-
606 int A_st = block_partitions[block_idx + 0];
-
607 int A_ed = block_partitions[block_idx + 1];
-
608 int B_st =
min (size_sorted_axis, 2 * sort_st + sort_sz / 2 + sort_md - A_st);
-
-
-
611 2 * sort_st + sort_sz / 2 + sort_md + sort_kernel::N_PER_BLOCK - A_ed);
-
-
613 if ((block_idx % merge_tiles) == merge_tiles - 1) {
-
614 A_ed =
min (size_sorted_axis, sort_st + sort_sz / 2);
-
615 B_ed =
min (size_sorted_axis, sort_st + sort_sz);
-
-
-
618 int A_sz = A_ed - A_st;
-
619 int B_sz = B_ed - B_st;
-
-
-
622 thread val_t thread_vals[N_PER_THREAD];
-
623 thread idx_t thread_idxs[N_PER_THREAD];
-
624 for (
int i = 0; i < N_PER_THREAD; i++) {
-
625 int idx = BLOCK_THREADS * i + lid.x;
-
626 if (idx < (A_sz + B_sz)) {
-
627 thread_vals[i] = (idx < A_sz) ? dev_vals_in[A_st + idx]
-
628 : dev_vals_in[B_st + idx - A_sz];
-
629 thread_idxs[i] = (idx < A_sz) ? dev_idxs_in[A_st + idx]
-
630 : dev_idxs_in[B_st + idx - A_sz];
-
-
632 thread_vals[i] = CompareOp::init;
-
-
-
-
-
-
638 threadgroup val_t tgp_vals[sort_kernel::N_PER_BLOCK];
-
639 threadgroup idx_t tgp_idxs[sort_kernel::N_PER_BLOCK];
-
640 threadgroup_barrier(mem_flags::mem_threadgroup);
-
641 for (
int i = 0; i < N_PER_THREAD; i++) {
-
642 int idx = BLOCK_THREADS * i + lid.x;
-
643 tgp_vals[idx] = thread_vals[i];
-
644 tgp_idxs[idx] = thread_idxs[i];
-
+
606 int block_idx = tid.x;
+
607 int merge_group = block_idx / merge_tiles;
+
608 int sort_st = sort_kernel::N_PER_BLOCK * merge_tiles * merge_group;
+
609 int sort_sz = sort_kernel::N_PER_BLOCK * merge_tiles;
+
610 int sort_md = sort_kernel::N_PER_BLOCK * block_idx - sort_st;
+
+
612 int A_st = block_partitions[block_idx + 0];
+
613 int A_ed = block_partitions[block_idx + 1];
+
614 int B_st =
min (size_sorted_axis, 2 * sort_st + sort_sz / 2 + sort_md - A_st);
+
+
+
617 2 * sort_st + sort_sz / 2 + sort_md + sort_kernel::N_PER_BLOCK - A_ed);
+
+
619 if ((block_idx % merge_tiles) == merge_tiles - 1) {
+
620 A_ed =
min (size_sorted_axis, sort_st + sort_sz / 2);
+
621 B_ed =
min (size_sorted_axis, sort_st + sort_sz);
+
+
+
624 int A_sz = A_ed - A_st;
+
625 int B_sz = B_ed - B_st;
+
+
+
628 thread val_t thread_vals[N_PER_THREAD];
+
629 thread idx_t thread_idxs[N_PER_THREAD];
+
630 for (
int i = 0; i < N_PER_THREAD; i++) {
+
631 int idx = BLOCK_THREADS * i + lid.x;
+
632 if (idx < (A_sz + B_sz)) {
+
633 thread_vals[i] = (idx < A_sz) ? dev_vals_in[A_st + idx]
+
634 : dev_vals_in[B_st + idx - A_sz];
+
635 thread_idxs[i] = (idx < A_sz) ? dev_idxs_in[A_st + idx]
+
636 : dev_idxs_in[B_st + idx - A_sz];
+
+
638 thread_vals[i] = CompareOp::init;
+
+
+
+
+
+
644 threadgroup val_t tgp_vals[sort_kernel::N_PER_BLOCK];
+
645 threadgroup idx_t tgp_idxs[sort_kernel::N_PER_BLOCK];
646 threadgroup_barrier(mem_flags::mem_threadgroup);
-
-
-
649 int sort_md_local =
min (A_sz + B_sz, N_PER_THREAD *
int (lid.x));
-
-
651 int A_st_local = block_sort_t::merge_partition(
-
652 tgp_vals, tgp_vals + A_sz, A_sz, B_sz, sort_md_local);
-
653 int A_ed_local = A_sz;
-
-
655 int B_st_local = sort_md_local - A_st_local;
-
656 int B_ed_local = B_sz;
-
-
658 int A_sz_local = A_ed_local - A_st_local;
-
659 int B_sz_local = B_ed_local - B_st_local;
+
647 for (
int i = 0; i < N_PER_THREAD; i++) {
+
648 int idx = BLOCK_THREADS * i + lid.x;
+
649 tgp_vals[idx] = thread_vals[i];
+
650 tgp_idxs[idx] = thread_idxs[i];
+
+
652 threadgroup_barrier(mem_flags::mem_threadgroup);
+
+
+
655 int sort_md_local =
min (A_sz + B_sz, N_PER_THREAD *
int (lid.x));
+
+
657 int A_st_local = block_sort_t::merge_partition(
+
658 tgp_vals, tgp_vals + A_sz, A_sz, B_sz, sort_md_local);
+
659 int A_ed_local = A_sz;
-
-
662 block_sort_t::merge_step(
-
663 tgp_vals + A_st_local,
-
664 tgp_vals + A_ed_local + B_st_local,
-
665 tgp_idxs + A_st_local,
-
666 tgp_idxs + A_ed_local + B_st_local,
-
-
-
-
-
-
672 threadgroup_barrier(mem_flags::mem_threadgroup);
-
673 for (
int i = 0; i < N_PER_THREAD; ++i) {
-
674 int idx = lid.x * N_PER_THREAD;
-
675 tgp_vals[idx + i] = thread_vals[i];
-
676 tgp_idxs[idx + i] = thread_idxs[i];
-
-
-
679 threadgroup_barrier(mem_flags::mem_threadgroup);
-
-
681 int base_idx = tid.x * sort_kernel::N_PER_BLOCK;
-
682 for (
int i = lid.x; i < sort_kernel::N_PER_BLOCK; i += BLOCK_THREADS) {
-
683 int idx = base_idx + i;
-
684 if (idx < size_sorted_axis) {
-
685 dev_vals_out[idx] = tgp_vals[i];
-
686 dev_idxs_out[idx] = tgp_idxs[i];
-
-
-
+
661 int B_st_local = sort_md_local - A_st_local;
+
662 int B_ed_local = B_sz;
+
+
664 int A_sz_local = A_ed_local - A_st_local;
+
665 int B_sz_local = B_ed_local - B_st_local;
+
+
+
668 block_sort_t::merge_step(
+
669 tgp_vals + A_st_local,
+
670 tgp_vals + A_ed_local + B_st_local,
+
671 tgp_idxs + A_st_local,
+
672 tgp_idxs + A_ed_local + B_st_local,
+
+
+
+
+
+
678 threadgroup_barrier(mem_flags::mem_threadgroup);
+
679 for (
int i = 0; i < N_PER_THREAD; ++i) {
+
680 int idx = lid.x * N_PER_THREAD;
+
681 tgp_vals[idx + i] = thread_vals[i];
+
682 tgp_idxs[idx + i] = thread_idxs[i];
+
+
+
685 threadgroup_barrier(mem_flags::mem_threadgroup);
+
+
687 int base_idx = tid.x * sort_kernel::N_PER_BLOCK;
+
688 for (
int i = lid.x; i < sort_kernel::N_PER_BLOCK; i += BLOCK_THREADS) {
+
689 int idx = base_idx + i;
+
690 if (idx < size_sorted_axis) {
+
691 dev_vals_out[idx] = tgp_vals[i];
+
692 dev_idxs_out[idx] = tgp_idxs[i];
+
+
+
Op op
Definition binary.h:141
@@ -812,10 +818,10 @@ $(function() { codefold.init(0); });
#define MLX_MTL_CONST
Definition sort.h:3
void block_sort_nc(const device T *inp, device U *out, const constant int &size_sorted_axis, const constant int &in_stride_sorted_axis, const constant int &out_stride_sorted_axis, const constant int &nc_dim, const device int *nc_shape, const device size_t *in_nc_strides, const device size_t *out_nc_strides, uint3 tid, uint3 lid)
Definition sort.h:338
void mb_block_sort(const device val_t *inp, device val_t *out_vals, device idx_t *out_idxs, const constant int &size_sorted_axis, const constant int &stride_sorted_axis, const constant int &nc_dim, const device int *nc_shape, const device size_t *nc_strides, uint3 tid, uint3 lid)
Definition sort.h:481
-
void mb_block_partition(device idx_t *block_partitions, const device val_t *dev_vals, const device idx_t *dev_idxs, const constant int &size_sorted_axis, const constant int &merge_tiles, uint3 tid, uint3 lid, uint3 tgp_dims)
Definition sort.h:526
+
void mb_block_partition(device idx_t *block_partitions, const device val_t *dev_vals, const device idx_t *dev_idxs, const constant int &size_sorted_axis, const constant int &merge_tiles, const constant int &n_blocks, uint3 tid, uint3 lid, uint3 tgp_dims)
Definition sort.h:525
METAL_FUNC void thread_swap(thread T &a, thread T &b)
Definition sort.h:16
void block_sort(const device T *inp, device U *out, const constant int &size_sorted_axis, const constant int &in_stride_sorted_axis, const constant int &out_stride_sorted_axis, const constant int &in_stride_segment_axis, const constant int &out_stride_segment_axis, uint3 tid, uint3 lid)
Definition sort.h:283
-
void mb_block_merge(const device idx_t *block_partitions, const device val_t *dev_vals_in, const device idx_t *dev_idxs_in, device val_t *dev_vals_out, device idx_t *dev_idxs_out, const constant int &size_sorted_axis, const constant int &merge_tiles, const constant int &num_tiles, uint3 tid, uint3 lid)
Definition sort.h:573
+
void mb_block_merge(const device idx_t *block_partitions, const device val_t *dev_vals_in, const device idx_t *dev_idxs_in, device val_t *dev_vals_out, device idx_t *dev_idxs_out, const constant int &size_sorted_axis, const constant int &merge_tiles, const constant int &num_tiles, uint3 tid, uint3 lid)
Definition sort.h:579
constant constexpr const int zero_helper
Definition sort.h:330
#define MLX_MTL_LOOP_UNROLL
Definition sort.h:4
diff --git a/docs/build/html/usage/compile.html b/docs/build/html/usage/compile.html
index d3769091b4..4b27a2ea64 100644
--- a/docs/build/html/usage/compile.html
+++ b/docs/build/html/usage/compile.html
@@ -8,7 +8,7 @@
-
Compilation — MLX 0.16.0 documentation
+
Compilation — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
-
+
@@ -131,8 +131,8 @@
-
-
+
+
@@ -296,6 +296,8 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod
+
mlx.core.einsum
+
mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv
@@ -343,6 +345,7 @@
mlx.core.minimum mlx.core.moveaxis mlx.core.multiply
+
mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones
@@ -414,6 +417,7 @@
mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform
+
mlx.core.random.laplace Transforms
diff --git a/docs/build/html/usage/distributed.html b/docs/build/html/usage/distributed.html
index 86997318b5..791666a087 100644
--- a/docs/build/html/usage/distributed.html
+++ b/docs/build/html/usage/distributed.html
@@ -8,7 +8,7 @@
Distributed Communication — MLX 0.16.0 documentation
+ Distributed Communication — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace Transforms
diff --git a/docs/build/html/usage/function_transforms.html b/docs/build/html/usage/function_transforms.html
index e8fc660981..a8ceb1a59d 100644
--- a/docs/build/html/usage/function_transforms.html
+++ b/docs/build/html/usage/function_transforms.html
@@ -8,7 +8,7 @@
Function Transforms — MLX 0.16.0 documentation
+ Function Transforms — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/usage/indexing.html b/docs/build/html/usage/indexing.html
index b2c74b7d39..4a31c2bb55 100644
--- a/docs/build/html/usage/indexing.html
+++ b/docs/build/html/usage/indexing.html
@@ -8,7 +8,7 @@
Indexing Arrays — MLX 0.16.0 documentation
+ Indexing Arrays — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/usage/lazy_evaluation.html b/docs/build/html/usage/lazy_evaluation.html
index 34cb87a974..0fa2d98ad6 100644
--- a/docs/build/html/usage/lazy_evaluation.html
+++ b/docs/build/html/usage/lazy_evaluation.html
@@ -8,7 +8,7 @@
Lazy Evaluation — MLX 0.16.0 documentation
+ Lazy Evaluation — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/usage/numpy.html b/docs/build/html/usage/numpy.html
index 89dc998a6e..e20516d6e8 100644
--- a/docs/build/html/usage/numpy.html
+++ b/docs/build/html/usage/numpy.html
@@ -8,7 +8,7 @@
Conversion to NumPy and Other Frameworks — MLX 0.16.0 documentation
+ Conversion to NumPy and Other Frameworks — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/usage/quick_start.html b/docs/build/html/usage/quick_start.html
index a34eb13944..c1aaad10df 100644
--- a/docs/build/html/usage/quick_start.html
+++ b/docs/build/html/usage/quick_start.html
@@ -8,7 +8,7 @@
Quick Start Guide — MLX 0.16.0 documentation
+ Quick Start Guide — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/usage/saving_and_loading.html b/docs/build/html/usage/saving_and_loading.html
index fc8c4c7920..87a9bd8df4 100644
--- a/docs/build/html/usage/saving_and_loading.html
+++ b/docs/build/html/usage/saving_and_loading.html
@@ -8,7 +8,7 @@
Saving and Loading Arrays — MLX 0.16.0 documentation
+ Saving and Loading Arrays — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/usage/unified_memory.html b/docs/build/html/usage/unified_memory.html
index c8d2be25da..145dcfae5f 100644
--- a/docs/build/html/usage/unified_memory.html
+++ b/docs/build/html/usage/unified_memory.html
@@ -8,7 +8,7 @@
Unified Memory — MLX 0.16.0 documentation
+ Unified Memory — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
diff --git a/docs/build/html/usage/using_streams.html b/docs/build/html/usage/using_streams.html
index 63dbe764b9..cd8f55cd39 100644
--- a/docs/build/html/usage/using_streams.html
+++ b/docs/build/html/usage/using_streams.html
@@ -8,7 +8,7 @@
Using Streams — MLX 0.16.0 documentation
+ Using Streams — MLX 0.16.1 documentation
@@ -36,7 +36,7 @@
mlx.core.diagonal mlx.core.divide mlx.core.divmod mlx.core.einsum mlx.core.einsum_path mlx.core.equal mlx.core.erf mlx.core.erfinv mlx.core.minimum mlx.core.moveaxis mlx.core.multiply mlx.core.nan_to_num mlx.core.negative mlx.core.not_equal mlx.core.ones mlx.core.random.split mlx.core.random.truncated_normal mlx.core.random.uniform mlx.core.random.laplace
Transforms
-
+ 
-
+ 
+
+
+
+
-
+