+ 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 ca9642f7b..f01a48434 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 cf312c483..d0dc8a547 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 33a12eef8..77f80e27f 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 d03ecb6cc..c686f1133 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 baf4abf42..56718a766 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 6f0603b89..9e2ad809f 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 5f2e0acf5..9ebdc17cb 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 199b63531..252fe28de 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 03fe336e2..26a634c2a 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 eb747a5c4..04f7386cc 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 6aca6e7e5..b635342df 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 8dae08447..19a9978c5 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 bf48b1f1d..ea3d8e660 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 c2cfacb3c..a4c035edb 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 2d492d9bf..055311582 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 832353618..745e6def3 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 76c1b62ba..73cd68f78 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 6bc2d9cb7..3f9a8ece1 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 c6c266af7..63c198c10 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 d56138996..5f0ae28cd 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 39e146781..ac19f10a8 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 2d8b62b7c..71dcecc2a 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 3a451f8a6..872e4509e 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 5014b4ee3..632124d02 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 07407d86c..a30ebb2e3 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 33d0ffc2f..dc0e76884 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 5e0d9e6eb..bcb09c603 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 bec4b214e..0d508b34b 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 df12f8705..57b35b4b0 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 67dbdcdf3..fafad459f 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 fb30af820..e202884e8 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 8b141d03c..9a2244928 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 4c38f5aa9..8355275cc 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 25e5fbd27..aa3fb6ecc 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 c1dbdacd9..2ba2ab0a1 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 cc4ef99b8..d48650f47 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 8fb72eb81..d34450194 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 520f9e7d0..0d03f7d24 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 25c34c963..af73d2a86 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 7b86224d8..68b9e8c48 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 f4d5d58f8..2bb84828c 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 bbdb631e0..6b1402dfc 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 ceafb2cf8..0a0a062df 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 a631f6228..dd434a0e5 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 ad8867a55..b7177e450 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 7f6f85b23..6a592025b 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 116013d18..9f0180289 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 7ece2ce28..05e25fa39 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 46ec256de..ee072eb71 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 59997b23c..60b0d8803 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 b5a76e725..fc9b234fb 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 95793fd1a..9dfa565d9 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 26e9c2004..f9c08d4e5 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 0165d7746..e69493bc1 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 6e797d83c..5f49d97d6 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 1fd0b0053..d1aaa4da0 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 780166334..555f42b38 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 47d4e52d1..235265153 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 f60059a76..bf694cda4 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 4cc214d67..ef54ad6b8 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 32d4049f5..a2f677a6c 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 c509d94c0..8aee0638c 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 cc74417c0..5f447c7e5 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 f06c9a471..d9564129a 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 9457bcc8a..b702172f7 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 091c2eabf..b087ee92e 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 91b6c4cce..7ce553bf9 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 584cae5b6..ba947b71e 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 ecb1083f0..9bd63f45c 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 25394470b..d9eb275d8 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 ea90e22bf..5c825776d 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 1c7ecd0ea..fbcfb302e 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 c44109836..d01f75ef6 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 56877909e..e230633f3 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 22788236c..7a95e96f8 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 8afeaf262..49f35c0ae 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 065f85990..aec628a4c 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 c93b1970c..4e0c9a9c8 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 5c5b0a235..232760d85 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 e15a58ad2..7cb7ccc43 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 6f0c782e4..6c1e7f840 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 69ca026c7..38f5fc8c6 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 efe3561ed..303647e19 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 f203ddc30..866979a34 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 5a568fa72..bca6bb2c2 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 5ed28a1ca..88f4b6c37 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 974159da7..26ebae2f8 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 2c128e1a4..f4bf1351d 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 8550aafa7..629625594 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 c06429b02..19a1870e0 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 51063843d..fdd1b8458 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 43b1a30f2..2ba021b08 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 68bed456f..ebcb72beb 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 472747e52..3785f3960 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 f21f89efd..7e43552d0 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 6bf121e3c..7b90b59e1 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 4a2ccbf28..38d5c5cef 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 3cbd503b5..ee0b125d6 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 c35526ea4..724855bc0 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 33e2f8082..dfc8231fe 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 629972226..c4b0a295d 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 991b23c98..7d7e52573 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 88fdddbc4..71b0b064a 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 fb98de44b..b1f637a72 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 60a41cd51..3c07cbe64 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 48167d2d5..fc4f3a1af 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 6e3d42e69..f52dfa57c 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 dbedc8371..3280cce3f 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 8118ce6d2..839d448e2 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 bdd32afa0..a7414d144 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 7482de7e0..bc2e80280 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 cc5b87605..c387f8b7e 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 b3dd36be2..f13e4d11f 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 0b19167ee..abb0566bc 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 eed06eaf9..f465f25bc 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 9a95b4751..997249e91 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 3fe6679ad..c64637e42 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 a6be6bd51..0944c9da5 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 a9fb59b7c..bc506704a 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 7ae680e39..70f90d7f4 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 01b576c11..7011b396c 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 f59268a16..41310b195 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 71139e207..fc300c15d 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 83b6278a8..cf49cfa37 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 2064603c8..29a300697 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 3eddcadbe..b41acbd92 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 aae55b83c..2b09a2a93 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 fb33f5b57..ac5e8586e 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 abc617ba0..b17ade634 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 7986e2ae8..0abeaaaac 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 bf82976a4..2ccfa05db 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 f903e5813..99f424285 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 a441cccef..e3117d7fd 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 0bc572bad..e9fa8907c 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 0c761eed3..7527b25c0 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 c062d0380..3b5a7260e 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 5e712c05b..31c479928 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 ee365addd..e68445d69 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 9c782b6a6..19467af45 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 26107917c..1e9c1e39d 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 db56af0de..bd9e77e16 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 f67aac8ba..13a9880c1 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 3c8258e1e..6b82fec83 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 4fb2f4775..4e61d06eb 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 c040e604d..f53bed063 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 78afd77bf..cdcd971dc 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 f0277df65..e08e8d56b 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 c3873fcc2..a3421dd44 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 891be1daa..3b2418786 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 009bfcf7a..1740be943 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 fefc8c27d..939a2aa28 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 cf40ab2e1..5319360df 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 6d58432a2..06fb95c62 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 48ed27592..553dc9c48 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 70ff4ff29..81467e026 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 513664da4..0d79265c6 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 485f74831..0aa78ae88 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 11753f077..fab8c4988 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 0c34b8195..54faffcbc 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 000000000..047801765
--- /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 a11088388..7089fee9c 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 28b91a1fb..0363830dc 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 c5fc69540..69340a1b2 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 afca0a63c..98d9d0ccb 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 c7b10baac..bb0ebb315 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 439bbe0ff..966c1904f 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 b01a71d98..d5873250d 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 65e78acf0..c933eddc6 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 fb92e1fa5..3f4553115 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 4596fca46..eee8bf0a2 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 9d575a10d..6b1e3aff8 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 96a787de3..5dc0aefb1 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 dc5a5cdba..269220e58 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 699811ef9..c63c24c72 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 a7815c5a6..2c1832af6 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 000000000..ce6b7ae40
--- /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 f70d04bad..e30c9b9b7 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 2f60543fb..31bea4854 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 5d0d923c2..28fff028d 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 73fb23cd8..e8d414356 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 67d3c2507..70190e111 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 3b70253e0..846475fe4 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 c90ad56f0..5edee69d8 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 13d76a606..8873222ff 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 32ef57124..b8af5997e 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 96eb6702a..3331984d2 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 27fe938d7..987bb3d56 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 b3a334a35..0c3a24901 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 fba34f023..6e016a10e 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 db8010186..ce39c2d29 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 b2735d749..e6f18aa61 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 2992c2d81..5d9973b57 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 c58c47b67..16fccc69e 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 08d930487..795ba4610 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 07de931a4..64c9be57d 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 e5c61c6f8..cb5587dd2 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 7493be161..d48df9a3c 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 4ec0226a4..e179809cf 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 735b6adf5..f83441c7e 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 a4b6744bb..a71532717 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 59eb31124..b7cf82e1b 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 61e60335b..a6ce38900 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 6af88f8b1..5741288a0 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 84a6677a8..76f1a44c4 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 ca0188c71..88fc1d71a 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 4b36f58c1..96ac5fc6c 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 9153e0861..0e1ee5626 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 cd670ee2e..c6fc13ce2 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 fe5480ea9..340c24ae1 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 bd158b0fc..76cfba25a 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 519df1bcf..1d27ba156 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 bdebd603d..638e66906 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 65ff4f10e..f1a294edd 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 ecce634ea..d83d7c52a 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 2f5aa7866..35ad16308 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 6f6fcb924..d301b1b05 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 7ef6db273..b63a8dafb 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 13371e0d5..f813b6efe 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 fbba22c63..204da0eeb 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 a310c3dbe..96a67b873 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 b6e3b6425..0ad65fd08 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 0f67ec515..292a4eab6 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 8f84edc00..5fa057f2c 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 47e290e90..16f0ef3b0 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 04bd02fbb..503887fcc 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 880319b32..5797233a1 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 def1ef307..0f1ce8422 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 75f41da6a..2b0710569 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 2dcfed998..adc4890b4 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 9ad1a3201..5a835b047 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 a68d69198..d5a05d21b 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 627f78b33..585d02447 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 1fdb1bba8..b036f11c5 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 37861975f..8103ba633 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 705d1603d..8177c05af 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 29d083106..4061036d3 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 0475348a4..648d38f57 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 44fc2e2b7..809cbbf7c 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 cf884f50a..c5ea43dfc 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 a7c64af2f..ea41c62f6 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 82b28a9c3..db79fc77f 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 b2de85315..5539dcc17 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 5ad873162..0b3d4c707 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 bee5cfd9d..799b71d3c 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 1e75f02cb..03a198871 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 0574e1953..a41552f2b 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 4577ea4f2..4eed55b9e 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 e07acd8f5..70c2caa76 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 35744da58..87b573c44 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 8313190e5..9fef328b3 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 166c3d787..4701a2107 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 b3e1247f4..3f446ae81 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 aae6a9ab2..c86948f86 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 8f2f98e09..4184a5a59 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 66847ebf2..2dcf9fb1c 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 815917945..cd6263d12 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 f9ad1726a..41f853371 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 039f2fbda..54b7dc464 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 e30e287ba..0016ee76d 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 c99c6ae3c..9ad90f14c 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 aa0e25034..731f2a1de 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 43ea7e513..752b79a76 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 1790ace0b..7482a45ed 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 c8e0676ce..98e5dacca 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 7f3579f29..459c534a4 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 58faa0500..5e68b69d1 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 e788c6956..a32e8e258 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 dd27fa81d..a534dfeb9 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 cca02150e..812a3399d 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 775f7b5d0..38cee3075 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 2b89db35a..4dc306c6f 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 bde4783ac..a5380327e 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 24a1fc1f1..5d752df9e 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 a598c7067..dfde82933 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 60bed5083..6ffb556a3 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 4dfa4a5b2..8ece62f0a 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 0e8d36985..9f7fd604b 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 aabac5df5..363ebaf21 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 295044fc7..6448d7c29 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 f36035cd9..6c3289c37 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 df2e1a9fa..b0342998b 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 34436d085..889d7017c 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 f22c57ced..3e8c2b0e3 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 df26e9ad8..b6af2babf 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 67d26fdee..7c931d5c6 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 22a5c226e..6ba8dc18d 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 39526fbb7..0da0dfe00 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 7f2777a52..7dbc3dfa2 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 e16813c95..aceaac1cf 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 535014516..714b1c0ad 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 d19ae49b3..587bf0e32 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 a54781cf2..e7eab13a6 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 8a0418559..87faba7f5 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 64e877122..78952beea 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 6c5ef1f21..18a901189 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 a0b94ade6..6e7c2c59b 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 947b75963..e46aba950 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 173b4c7d6..d5f1956e4 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 632c72884..c70e85640 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 2bcd70645..b732708cc 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 f54dce682..80d40d186 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 766435658..99fd5062d 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 1206f3f22..6d1bd4590 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 6f091c151..f06fc7458 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 eae7b2e38..6f81764c3 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 2867372da..6f3061c97 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 98dfb5b72..78c610cf1 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 0ed489508..43ee457e0 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 8f9e9ee0b..db214eb14 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 37b9180a3..c8c81f989 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 a1a922b2d..071a7155e 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 c8302a461..2a2d05d0b 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 94933e07c..4d8ec244c 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 ef8a954e0..b12cfbac6 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 e5bbe468f..14a7f6b39 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 1a15abb02..7af8bdd26 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 51305f472..63b0a9b60 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 7485ea89b..2d76d4bec 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 24e73687d..825eecd5c 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 085c4b380..f22971cc0 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 199bc1c6c..8f04b61b7 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 ec0766b6a..258979d76 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 46b8faec6..ed82d0e6d 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 c3ef9ce4f..1d6dba783 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 c4606219e..f8392f6f5 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 538c907a6..d98a9b4de 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 c30e0215c..345a0e405 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 227bff6c2..771f9daef 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 54656f843..992d1cc10 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 b24fa24b3..5a6478b2b 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 de6f14e02..f1d32e935 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 62bea8396..2082622cc 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 b5fee83b7..f10ca6d11 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 9c2762f0c..b10b2e267 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 782438392..e8fa10d08 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 12ba32099..c8844ece7 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 e204d7a1a..0500a996d 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 70223c02d..ee92ba858 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 cd46682d4..21c2255e4 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 826e478d3..3a1e83d79 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 22bc6dae2..5aea944b7 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 8d3e60ed3..c377bf25c 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 3b6183c43..f500ba8d0 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 6c4416677..43061455f 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 b961a5d2b..7c9e6ce48 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 f70d963d1..8a7af2f75 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 794c10398..435976e5c 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 560e13de0..76093ff23 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 e2ae6be03..f5a8c26ce 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 9a9a62b91..e71fdd294 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 eaf286b49..f1f6f0785 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 1b7b48ff6..7918dc722 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 0808c23ef..760b04d3a 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 f3b43fce2..fd702a7be 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 167802cc2..845f21488 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 d05f4c995..aa65e46a1 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 0006e0dec..fe0e6fa1d 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 e9713bc71..e8f2fd38e 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 639268b6e..ac4c9990a 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 001ec10cd..99cfef2e3 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 e03640c3c..a0ef6a7f4 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 4aaeaa6a0..d47ac4b04 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 cf1b1507d..11573a5d6 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 9acc3c8d4..f65c6caeb 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 b0eb5bb29..1c08a6569 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 653770740..e5072b9f7 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 fc4be847c..87527034d 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 f0daede6d..546c3b8aa 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 d2420a300..9c1569a58 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 9a5d2ceb3..9e55aa224 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 3ab860bbb..ef6e72ac8 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 9b34ea83f..e305f9f31 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 6e75250ff..3402bf78a 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 5d185d1cc..834bb80bd 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 ac978b275..295e9788a 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 ac0add56f..6bd5e76bb 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 4e82436ab..d59b3227a 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 2e0590ae9..01a13c1d3 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 d07f36764..4acd664a3 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 faf853f96..ed5854ec5 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 65ae4f2e0..7f60ac15f 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 e47c617dd..1bebc60c3 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 bf1bb29de..c5ef73578 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 f9c5e0608..8b9ee6ebd 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 a891513be..5ab5faaeb 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 6e596cdb2..10418ec70 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 ac02737c2..56fea43f2 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 84025260e..a9cb0de38 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 159ac43a2..90e93be58 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 8ac2f33e0..44516b1cb 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 d3769091b..4b27a2ea6 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 86997318b..791666a08 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 e8fc66098..a8ceb1a59 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 b2c74b7d3..4a31c2bb5 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 34cb87a97..0fa2d98ad 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 89dc998a6..e20516d6e 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 a34eb1394..c1aaad10d 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 fc8c4c792..87a9bd8df 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 c8d2be25d..145dcfae5 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 63dbe764b..cd8f55cd3 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
-
+ 
-
+ 
+
+
+
+
-
+