diff --git a/docs/build/doctrees/cpp/ops.doctree b/docs/build/doctrees/cpp/ops.doctree deleted file mode 100644 index 8ade670227..0000000000 Binary files a/docs/build/doctrees/cpp/ops.doctree and /dev/null differ diff --git a/docs/build/doctrees/dev/extensions.doctree b/docs/build/doctrees/dev/extensions.doctree deleted file mode 100644 index a0d2baf1d8..0000000000 Binary files a/docs/build/doctrees/dev/extensions.doctree and /dev/null differ diff --git a/docs/build/doctrees/environment.pickle b/docs/build/doctrees/environment.pickle deleted file mode 100644 index 2f7d964932..0000000000 Binary files a/docs/build/doctrees/environment.pickle and /dev/null differ diff --git a/docs/build/doctrees/examples/linear_regression.doctree b/docs/build/doctrees/examples/linear_regression.doctree deleted file mode 100644 index 4b9be359ee..0000000000 Binary files a/docs/build/doctrees/examples/linear_regression.doctree and /dev/null differ diff --git a/docs/build/doctrees/examples/llama-inference.doctree b/docs/build/doctrees/examples/llama-inference.doctree deleted file mode 100644 index 23804ca7a4..0000000000 Binary files a/docs/build/doctrees/examples/llama-inference.doctree and /dev/null differ diff --git a/docs/build/doctrees/examples/mlp.doctree b/docs/build/doctrees/examples/mlp.doctree deleted file mode 100644 index 20decac880..0000000000 Binary files a/docs/build/doctrees/examples/mlp.doctree and /dev/null differ diff --git a/docs/build/doctrees/index.doctree b/docs/build/doctrees/index.doctree deleted file mode 100644 index 96bbda118e..0000000000 Binary files a/docs/build/doctrees/index.doctree and /dev/null differ diff --git a/docs/build/doctrees/install.doctree b/docs/build/doctrees/install.doctree deleted file mode 100644 index 8d5f888092..0000000000 Binary files a/docs/build/doctrees/install.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.Device.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.Device.doctree deleted file mode 100644 index 0e8b538370..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.Device.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.Dtype.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.Dtype.doctree deleted file mode 100644 index b76aa61003..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.Dtype.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.Stream.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.Stream.doctree deleted file mode 100644 index b09770dda3..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.Stream.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.abs.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.abs.doctree deleted file mode 100644 index 8d6d057229..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.abs.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.add.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.add.doctree deleted file mode 100644 index f2286c5aaa..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.add.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.all.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.all.doctree deleted file mode 100644 index b8e2f1d379..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.all.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.allclose.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.allclose.doctree deleted file mode 100644 index 657f0c24bf..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.allclose.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.any.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.any.doctree deleted file mode 100644 index b7da1ff329..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.any.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.arange.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.arange.doctree deleted file mode 100644 index 0fd022e896..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.arange.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.arccos.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.arccos.doctree deleted file mode 100644 index f7ab0f64a7..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.arccos.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.arccosh.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.arccosh.doctree deleted file mode 100644 index e251f756c3..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.arccosh.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.arcsin.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.arcsin.doctree deleted file mode 100644 index 4eaeb99455..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.arcsin.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.arcsinh.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.arcsinh.doctree deleted file mode 100644 index d567a4b93d..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.arcsinh.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.arctan.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.arctan.doctree deleted file mode 100644 index 7e1681e422..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.arctan.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.arctanh.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.arctanh.doctree deleted file mode 100644 index 4203ad352b..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.arctanh.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.argmax.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.argmax.doctree deleted file mode 100644 index 7f3ef214fd..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.argmax.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.argmin.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.argmin.doctree deleted file mode 100644 index a85e59e58c..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.argmin.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.argpartition.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.argpartition.doctree deleted file mode 100644 index a309dea13f..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.argpartition.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.argsort.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.argsort.doctree deleted file mode 100644 index 17aaf5d4a4..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.argsort.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.array.T.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.array.T.doctree deleted file mode 100644 index 8a845dddcb..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.array.T.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.array.abs.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.array.abs.doctree deleted file mode 100644 index 9429798060..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.array.abs.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.array.all.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.array.all.doctree deleted file mode 100644 index 79f01e39de..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.array.all.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.array.any.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.array.any.doctree deleted file mode 100644 index 98eb5675f3..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.array.any.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.array.argmax.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.array.argmax.doctree deleted file mode 100644 index 78d192cbe3..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.array.argmax.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.array.argmin.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.array.argmin.doctree deleted file mode 100644 index ea20438611..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.array.argmin.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.array.astype.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.array.astype.doctree deleted file mode 100644 index 1dd1944a71..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.array.astype.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.array.cos.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.array.cos.doctree deleted file mode 100644 index 5903c60b38..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.array.cos.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.array.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.array.doctree deleted file mode 100644 index 43418f8c8b..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.array.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.array.dtype.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.array.dtype.doctree deleted file mode 100644 index 19bef20a79..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.array.dtype.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.array.exp.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.array.exp.doctree deleted file mode 100644 index 6a32c5103b..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.array.exp.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.array.item.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.array.item.doctree deleted file mode 100644 index ec88b9d46a..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.array.item.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.array.log.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.array.log.doctree deleted file mode 100644 index db73a238c7..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.array.log.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.array.log1p.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.array.log1p.doctree deleted file mode 100644 index 3840dcd764..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.array.log1p.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.array.logsumexp.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.array.logsumexp.doctree deleted file mode 100644 index 1ecb477abf..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.array.logsumexp.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.array.max.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.array.max.doctree deleted file mode 100644 index 261fdf7ba6..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.array.max.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.array.mean.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.array.mean.doctree deleted file mode 100644 index c14cb4fb05..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.array.mean.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.array.min.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.array.min.doctree deleted file mode 100644 index 9a739fba54..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.array.min.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.array.ndim.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.array.ndim.doctree deleted file mode 100644 index 12ff801921..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.array.ndim.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.array.prod.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.array.prod.doctree deleted file mode 100644 index 643d5471c9..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.array.prod.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.array.reciprocal.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.array.reciprocal.doctree deleted file mode 100644 index 3156e20206..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.array.reciprocal.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.array.reshape.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.array.reshape.doctree deleted file mode 100644 index f52a865118..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.array.reshape.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.array.round.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.array.round.doctree deleted file mode 100644 index 9afc4c91bb..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.array.round.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.array.rsqrt.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.array.rsqrt.doctree deleted file mode 100644 index 3c275626b0..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.array.rsqrt.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.array.shape.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.array.shape.doctree deleted file mode 100644 index 2f61834313..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.array.shape.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.array.sin.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.array.sin.doctree deleted file mode 100644 index 51752c2a90..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.array.sin.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.array.size.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.array.size.doctree deleted file mode 100644 index c73cad8694..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.array.size.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.array.split.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.array.split.doctree deleted file mode 100644 index 8a1f5e9e7b..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.array.split.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.array.sqrt.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.array.sqrt.doctree deleted file mode 100644 index 0045103e07..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.array.sqrt.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.array.square.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.array.square.doctree deleted file mode 100644 index 9ed6d6f8bc..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.array.square.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.array.sum.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.array.sum.doctree deleted file mode 100644 index d8a5adefad..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.array.sum.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.array.tolist.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.array.tolist.doctree deleted file mode 100644 index d7866eb135..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.array.tolist.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.array.transpose.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.array.transpose.doctree deleted file mode 100644 index f7922948f8..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.array.transpose.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.array.var.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.array.var.doctree deleted file mode 100644 index 1067c4261b..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.array.var.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.array_equal.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.array_equal.doctree deleted file mode 100644 index 77eec964ba..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.array_equal.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.broadcast_to.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.broadcast_to.doctree deleted file mode 100644 index 6dca61080e..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.broadcast_to.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.ceil.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.ceil.doctree deleted file mode 100644 index f55398aa27..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.ceil.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.clip.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.clip.doctree deleted file mode 100644 index bcf04a4973..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.clip.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.concatenate.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.concatenate.doctree deleted file mode 100644 index cdbf971101..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.concatenate.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.conv1d.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.conv1d.doctree deleted file mode 100644 index 77b1b8ca2c..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.conv1d.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.conv2d.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.conv2d.doctree deleted file mode 100644 index 9eacea9e03..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.conv2d.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.convolve.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.convolve.doctree deleted file mode 100644 index c4823f85cb..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.convolve.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.cos.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.cos.doctree deleted file mode 100644 index 035779f02a..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.cos.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.cosh.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.cosh.doctree deleted file mode 100644 index 474dfbf72a..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.cosh.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.default_device.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.default_device.doctree deleted file mode 100644 index f530d1da57..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.default_device.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.default_stream.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.default_stream.doctree deleted file mode 100644 index 7198c13a93..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.default_stream.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.dequantize.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.dequantize.doctree deleted file mode 100644 index 1e05a407d2..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.dequantize.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.divide.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.divide.doctree deleted file mode 100644 index 964b2400cf..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.divide.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.divmod.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.divmod.doctree deleted file mode 100644 index dfdfaca644..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.divmod.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.equal.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.equal.doctree deleted file mode 100644 index cd39a49c6a..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.equal.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.erf.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.erf.doctree deleted file mode 100644 index ddc2eefed4..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.erf.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.erfinv.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.erfinv.doctree deleted file mode 100644 index 1676d7dbc1..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.erfinv.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.eval.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.eval.doctree deleted file mode 100644 index e9a0ba3674..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.eval.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.exp.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.exp.doctree deleted file mode 100644 index a5515bc24c..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.exp.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.expand_dims.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.expand_dims.doctree deleted file mode 100644 index 2b4cc7a1ca..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.expand_dims.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.eye.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.eye.doctree deleted file mode 100644 index 5f93ca3e20..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.eye.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.fft.fft.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.fft.fft.doctree deleted file mode 100644 index 279ef64733..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.fft.fft.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.fft.fft2.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.fft.fft2.doctree deleted file mode 100644 index 406878c3cb..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.fft.fft2.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.fft.fftn.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.fft.fftn.doctree deleted file mode 100644 index 06fe68acc5..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.fft.fftn.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.fft.ifft.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.fft.ifft.doctree deleted file mode 100644 index c23ebb2cbd..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.fft.ifft.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.fft.ifft2.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.fft.ifft2.doctree deleted file mode 100644 index 3bc531ea87..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.fft.ifft2.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.fft.ifftn.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.fft.ifftn.doctree deleted file mode 100644 index 640ca7896a..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.fft.ifftn.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.fft.irfft.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.fft.irfft.doctree deleted file mode 100644 index 8372bf45e4..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.fft.irfft.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.fft.irfft2.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.fft.irfft2.doctree deleted file mode 100644 index 6cb1aaddf0..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.fft.irfft2.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.fft.irfftn.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.fft.irfftn.doctree deleted file mode 100644 index 253424bd44..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.fft.irfftn.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.fft.rfft.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.fft.rfft.doctree deleted file mode 100644 index 87e93b5446..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.fft.rfft.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.fft.rfft2.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.fft.rfft2.doctree deleted file mode 100644 index 1c61d4992c..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.fft.rfft2.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.fft.rfftn.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.fft.rfftn.doctree deleted file mode 100644 index 3c41914bf6..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.fft.rfftn.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.flatten.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.flatten.doctree deleted file mode 100644 index 841f020bd9..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.flatten.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.floor.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.floor.doctree deleted file mode 100644 index 94cfb529b8..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.floor.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.floor_divide.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.floor_divide.doctree deleted file mode 100644 index 8d63347ece..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.floor_divide.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.full.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.full.doctree deleted file mode 100644 index ca8e07c205..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.full.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.grad.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.grad.doctree deleted file mode 100644 index bd99443dcb..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.grad.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.greater.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.greater.doctree deleted file mode 100644 index 0e625e3426..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.greater.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.greater_equal.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.greater_equal.doctree deleted file mode 100644 index 5573eb5565..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.greater_equal.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.identity.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.identity.doctree deleted file mode 100644 index e11d9bd2db..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.identity.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.inner.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.inner.doctree deleted file mode 100644 index d48aa6bc8a..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.inner.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.isinf.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.isinf.doctree deleted file mode 100644 index 830ebb9a4f..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.isinf.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.isnan.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.isnan.doctree deleted file mode 100644 index 8322e6451c..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.isnan.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.isneginf.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.isneginf.doctree deleted file mode 100644 index 3cb9d80d6f..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.isneginf.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.isposinf.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.isposinf.doctree deleted file mode 100644 index aac5c0a6aa..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.isposinf.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.jvp.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.jvp.doctree deleted file mode 100644 index 7c6fbd1b16..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.jvp.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.less.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.less.doctree deleted file mode 100644 index f9e9576869..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.less.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.less_equal.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.less_equal.doctree deleted file mode 100644 index 808bf41e6e..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.less_equal.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.linalg.norm.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.linalg.norm.doctree deleted file mode 100644 index 4bf27e832e..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.linalg.norm.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.linspace.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.linspace.doctree deleted file mode 100644 index 03abea1138..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.linspace.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.load.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.load.doctree deleted file mode 100644 index eee3858aa3..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.load.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.log.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.log.doctree deleted file mode 100644 index 0b4027e64a..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.log.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.log10.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.log10.doctree deleted file mode 100644 index 28848956a8..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.log10.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.log1p.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.log1p.doctree deleted file mode 100644 index 91a06483e8..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.log1p.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.log2.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.log2.doctree deleted file mode 100644 index 62d2e0dad6..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.log2.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.logaddexp.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.logaddexp.doctree deleted file mode 100644 index 3b7cc020d2..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.logaddexp.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.logical_and.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.logical_and.doctree deleted file mode 100644 index 9dfc94bfda..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.logical_and.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.logical_not.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.logical_not.doctree deleted file mode 100644 index 43109d0cb7..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.logical_not.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.logical_or.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.logical_or.doctree deleted file mode 100644 index b6db9d22ab..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.logical_or.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.logsumexp.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.logsumexp.doctree deleted file mode 100644 index 2ecb372564..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.logsumexp.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.matmul.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.matmul.doctree deleted file mode 100644 index 31c0b2bd49..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.matmul.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.max.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.max.doctree deleted file mode 100644 index c13fe054f0..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.max.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.maximum.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.maximum.doctree deleted file mode 100644 index ef48dff78c..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.maximum.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.mean.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.mean.doctree deleted file mode 100644 index effa5331bf..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.mean.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.min.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.min.doctree deleted file mode 100644 index 78741b97a2..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.min.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.minimum.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.minimum.doctree deleted file mode 100644 index 8d836911fc..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.minimum.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.moveaxis.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.moveaxis.doctree deleted file mode 100644 index 1c39b81aec..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.moveaxis.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.multiply.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.multiply.doctree deleted file mode 100644 index c374fa7e7f..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.multiply.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.negative.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.negative.doctree deleted file mode 100644 index 13feb86553..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.negative.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.new_stream.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.new_stream.doctree deleted file mode 100644 index e9cf8d29e7..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.new_stream.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.ones.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.ones.doctree deleted file mode 100644 index 293e89f6c9..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.ones.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.ones_like.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.ones_like.doctree deleted file mode 100644 index 5576b0db85..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.ones_like.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.outer.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.outer.doctree deleted file mode 100644 index 43ac2a1d07..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.outer.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.pad.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.pad.doctree deleted file mode 100644 index de3a605fd0..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.pad.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.partition.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.partition.doctree deleted file mode 100644 index e70afcebcd..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.partition.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.prod.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.prod.doctree deleted file mode 100644 index baf59662e0..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.prod.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.quantize.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.quantize.doctree deleted file mode 100644 index cd5d60f56f..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.quantize.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.quantized_matmul.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.quantized_matmul.doctree deleted file mode 100644 index 28716d48a7..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.quantized_matmul.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.random.bernoulli.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.random.bernoulli.doctree deleted file mode 100644 index 51bd419315..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.random.bernoulli.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.random.categorical.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.random.categorical.doctree deleted file mode 100644 index 9cf4871c41..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.random.categorical.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.random.gumbel.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.random.gumbel.doctree deleted file mode 100644 index b49ceb730c..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.random.gumbel.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.random.key.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.random.key.doctree deleted file mode 100644 index 7c388509d9..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.random.key.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.random.normal.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.random.normal.doctree deleted file mode 100644 index c59a5f8aac..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.random.normal.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.random.randint.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.random.randint.doctree deleted file mode 100644 index c66cc8b907..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.random.randint.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.random.seed.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.random.seed.doctree deleted file mode 100644 index a8f54cc528..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.random.seed.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.random.split.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.random.split.doctree deleted file mode 100644 index fdbc46917a..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.random.split.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.random.truncated_normal.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.random.truncated_normal.doctree deleted file mode 100644 index 2319da0310..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.random.truncated_normal.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.random.uniform.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.random.uniform.doctree deleted file mode 100644 index 61200ae175..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.random.uniform.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.reciprocal.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.reciprocal.doctree deleted file mode 100644 index 7151f851aa..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.reciprocal.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.repeat.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.repeat.doctree deleted file mode 100644 index f65c41fa8e..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.repeat.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.reshape.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.reshape.doctree deleted file mode 100644 index f3f77d8567..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.reshape.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.round.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.round.doctree deleted file mode 100644 index 21f35145b8..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.round.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.rsqrt.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.rsqrt.doctree deleted file mode 100644 index 26bb22da05..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.rsqrt.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.save.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.save.doctree deleted file mode 100644 index 62c6b196ca..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.save.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.save_gguf.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.save_gguf.doctree deleted file mode 100644 index e5603bc97f..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.save_gguf.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.save_safetensors.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.save_safetensors.doctree deleted file mode 100644 index fde220212f..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.save_safetensors.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.savez.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.savez.doctree deleted file mode 100644 index 5bb76b6dc9..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.savez.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.savez_compressed.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.savez_compressed.doctree deleted file mode 100644 index 2363ee48af..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.savez_compressed.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.set_default_device.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.set_default_device.doctree deleted file mode 100644 index 5484b3d35c..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.set_default_device.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.set_default_stream.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.set_default_stream.doctree deleted file mode 100644 index 3b72c0325d..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.set_default_stream.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.sigmoid.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.sigmoid.doctree deleted file mode 100644 index 4acb22b290..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.sigmoid.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.sign.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.sign.doctree deleted file mode 100644 index 1a099b4ee8..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.sign.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.simplify.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.simplify.doctree deleted file mode 100644 index c8d8e4a8e0..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.simplify.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.sin.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.sin.doctree deleted file mode 100644 index de8e10ad5a..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.sin.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.sinh.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.sinh.doctree deleted file mode 100644 index acf0171e22..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.sinh.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.softmax.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.softmax.doctree deleted file mode 100644 index 5a5a74b3cf..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.softmax.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.sort.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.sort.doctree deleted file mode 100644 index d6aa3f72e5..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.sort.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.split.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.split.doctree deleted file mode 100644 index ee68048f01..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.split.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.sqrt.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.sqrt.doctree deleted file mode 100644 index 3c8c33a8b1..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.sqrt.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.square.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.square.doctree deleted file mode 100644 index 602fbe0235..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.square.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.squeeze.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.squeeze.doctree deleted file mode 100644 index 9b8a960b9a..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.squeeze.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.stack.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.stack.doctree deleted file mode 100644 index 96b5b740e5..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.stack.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.stop_gradient.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.stop_gradient.doctree deleted file mode 100644 index 2f5d6ba121..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.stop_gradient.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.subtract.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.subtract.doctree deleted file mode 100644 index 295b0f4122..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.subtract.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.sum.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.sum.doctree deleted file mode 100644 index 2997e42718..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.sum.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.swapaxes.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.swapaxes.doctree deleted file mode 100644 index dcde15d9f5..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.swapaxes.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.take.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.take.doctree deleted file mode 100644 index 8512009d95..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.take.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.take_along_axis.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.take_along_axis.doctree deleted file mode 100644 index b1644987b6..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.take_along_axis.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.tan.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.tan.doctree deleted file mode 100644 index fabce8aff2..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.tan.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.tanh.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.tanh.doctree deleted file mode 100644 index 38b040c871..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.tanh.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.tensordot.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.tensordot.doctree deleted file mode 100644 index 8112528035..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.tensordot.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.transpose.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.transpose.doctree deleted file mode 100644 index 994e06fcc5..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.transpose.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.tri.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.tri.doctree deleted file mode 100644 index c8e4f3593b..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.tri.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.tril.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.tril.doctree deleted file mode 100644 index 8920df3fbc..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.tril.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.triu.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.triu.doctree deleted file mode 100644 index 52fabce5dc..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.triu.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.value_and_grad.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.value_and_grad.doctree deleted file mode 100644 index c6a282a289..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.value_and_grad.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.var.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.var.doctree deleted file mode 100644 index 0513ecacc2..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.var.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.vjp.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.vjp.doctree deleted file mode 100644 index f89559aba8..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.vjp.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.vmap.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.vmap.doctree deleted file mode 100644 index 868898ddb8..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.vmap.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.where.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.where.doctree deleted file mode 100644 index 9cb374060e..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.where.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.zeros.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.zeros.doctree deleted file mode 100644 index e4499ae4ca..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.zeros.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.core.zeros_like.doctree b/docs/build/doctrees/python/_autosummary/mlx.core.zeros_like.doctree deleted file mode 100644 index 3682d97704..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.core.zeros_like.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.nn.value_and_grad.doctree b/docs/build/doctrees/python/_autosummary/mlx.nn.value_and_grad.doctree deleted file mode 100644 index 96a0ef2268..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.nn.value_and_grad.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.optimizers.AdaDelta.doctree b/docs/build/doctrees/python/_autosummary/mlx.optimizers.AdaDelta.doctree deleted file mode 100644 index 65030721aa..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.optimizers.AdaDelta.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.optimizers.Adagrad.doctree b/docs/build/doctrees/python/_autosummary/mlx.optimizers.Adagrad.doctree deleted file mode 100644 index 4b414542a6..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.optimizers.Adagrad.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.optimizers.Adam.doctree b/docs/build/doctrees/python/_autosummary/mlx.optimizers.Adam.doctree deleted file mode 100644 index e3291a8d14..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.optimizers.Adam.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.optimizers.AdamW.doctree b/docs/build/doctrees/python/_autosummary/mlx.optimizers.AdamW.doctree deleted file mode 100644 index cfe6a6c0d2..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.optimizers.AdamW.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.optimizers.Adamax.doctree b/docs/build/doctrees/python/_autosummary/mlx.optimizers.Adamax.doctree deleted file mode 100644 index 5bbe3c40b7..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.optimizers.Adamax.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.optimizers.Lion.doctree b/docs/build/doctrees/python/_autosummary/mlx.optimizers.Lion.doctree deleted file mode 100644 index 4fa8ddcc08..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.optimizers.Lion.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.optimizers.Optimizer.doctree b/docs/build/doctrees/python/_autosummary/mlx.optimizers.Optimizer.doctree deleted file mode 100644 index 921aca1558..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.optimizers.Optimizer.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.optimizers.OptimizerState.doctree b/docs/build/doctrees/python/_autosummary/mlx.optimizers.OptimizerState.doctree deleted file mode 100644 index d4a69be2e0..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.optimizers.OptimizerState.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.optimizers.RMSprop.doctree b/docs/build/doctrees/python/_autosummary/mlx.optimizers.RMSprop.doctree deleted file mode 100644 index 8f8379715c..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.optimizers.RMSprop.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.optimizers.SGD.doctree b/docs/build/doctrees/python/_autosummary/mlx.optimizers.SGD.doctree deleted file mode 100644 index 215c29f803..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.optimizers.SGD.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.utils.tree_flatten.doctree b/docs/build/doctrees/python/_autosummary/mlx.utils.tree_flatten.doctree deleted file mode 100644 index cab9354870..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.utils.tree_flatten.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.utils.tree_map.doctree b/docs/build/doctrees/python/_autosummary/mlx.utils.tree_map.doctree deleted file mode 100644 index e7351cc08f..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.utils.tree_map.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/_autosummary/mlx.utils.tree_unflatten.doctree b/docs/build/doctrees/python/_autosummary/mlx.utils.tree_unflatten.doctree deleted file mode 100644 index 703f7e65f4..0000000000 Binary files a/docs/build/doctrees/python/_autosummary/mlx.utils.tree_unflatten.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/array.doctree b/docs/build/doctrees/python/array.doctree deleted file mode 100644 index e365864471..0000000000 Binary files a/docs/build/doctrees/python/array.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/data_types.doctree b/docs/build/doctrees/python/data_types.doctree deleted file mode 100644 index d4ad20ae36..0000000000 Binary files a/docs/build/doctrees/python/data_types.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/devices_and_streams.doctree b/docs/build/doctrees/python/devices_and_streams.doctree deleted file mode 100644 index 6a30a134cd..0000000000 Binary files a/docs/build/doctrees/python/devices_and_streams.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/fft.doctree b/docs/build/doctrees/python/fft.doctree deleted file mode 100644 index 2bede7452a..0000000000 Binary files a/docs/build/doctrees/python/fft.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/linalg.doctree b/docs/build/doctrees/python/linalg.doctree deleted file mode 100644 index 87a09a6a93..0000000000 Binary files a/docs/build/doctrees/python/linalg.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/nn.doctree b/docs/build/doctrees/python/nn.doctree deleted file mode 100644 index 93ceef54f5..0000000000 Binary files a/docs/build/doctrees/python/nn.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/nn/_autosummary/mlx.nn.ALiBi.doctree b/docs/build/doctrees/python/nn/_autosummary/mlx.nn.ALiBi.doctree deleted file mode 100644 index ce1b0d9280..0000000000 Binary files a/docs/build/doctrees/python/nn/_autosummary/mlx.nn.ALiBi.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/nn/_autosummary/mlx.nn.BatchNorm.doctree b/docs/build/doctrees/python/nn/_autosummary/mlx.nn.BatchNorm.doctree deleted file mode 100644 index 556c5024fe..0000000000 Binary files a/docs/build/doctrees/python/nn/_autosummary/mlx.nn.BatchNorm.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/nn/_autosummary/mlx.nn.Conv1d.doctree b/docs/build/doctrees/python/nn/_autosummary/mlx.nn.Conv1d.doctree deleted file mode 100644 index 2f1d10293f..0000000000 Binary files a/docs/build/doctrees/python/nn/_autosummary/mlx.nn.Conv1d.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/nn/_autosummary/mlx.nn.Conv2d.doctree b/docs/build/doctrees/python/nn/_autosummary/mlx.nn.Conv2d.doctree deleted file mode 100644 index 779bcd0267..0000000000 Binary files a/docs/build/doctrees/python/nn/_autosummary/mlx.nn.Conv2d.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/nn/_autosummary/mlx.nn.Dropout.doctree b/docs/build/doctrees/python/nn/_autosummary/mlx.nn.Dropout.doctree deleted file mode 100644 index bc3c9ccc3b..0000000000 Binary files a/docs/build/doctrees/python/nn/_autosummary/mlx.nn.Dropout.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/nn/_autosummary/mlx.nn.Dropout2d.doctree b/docs/build/doctrees/python/nn/_autosummary/mlx.nn.Dropout2d.doctree deleted file mode 100644 index 9f172fb414..0000000000 Binary files a/docs/build/doctrees/python/nn/_autosummary/mlx.nn.Dropout2d.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/nn/_autosummary/mlx.nn.Dropout3d.doctree b/docs/build/doctrees/python/nn/_autosummary/mlx.nn.Dropout3d.doctree deleted file mode 100644 index 60a8d9a2fd..0000000000 Binary files a/docs/build/doctrees/python/nn/_autosummary/mlx.nn.Dropout3d.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/nn/_autosummary/mlx.nn.Embedding.doctree b/docs/build/doctrees/python/nn/_autosummary/mlx.nn.Embedding.doctree deleted file mode 100644 index 684a41cde9..0000000000 Binary files a/docs/build/doctrees/python/nn/_autosummary/mlx.nn.Embedding.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/nn/_autosummary/mlx.nn.GELU.doctree b/docs/build/doctrees/python/nn/_autosummary/mlx.nn.GELU.doctree deleted file mode 100644 index 2304bd122d..0000000000 Binary files a/docs/build/doctrees/python/nn/_autosummary/mlx.nn.GELU.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/nn/_autosummary/mlx.nn.GroupNorm.doctree b/docs/build/doctrees/python/nn/_autosummary/mlx.nn.GroupNorm.doctree deleted file mode 100644 index 9e7d9afd2e..0000000000 Binary files a/docs/build/doctrees/python/nn/_autosummary/mlx.nn.GroupNorm.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/nn/_autosummary/mlx.nn.InstanceNorm.doctree b/docs/build/doctrees/python/nn/_autosummary/mlx.nn.InstanceNorm.doctree deleted file mode 100644 index a6f01f3882..0000000000 Binary files a/docs/build/doctrees/python/nn/_autosummary/mlx.nn.InstanceNorm.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/nn/_autosummary/mlx.nn.LayerNorm.doctree b/docs/build/doctrees/python/nn/_autosummary/mlx.nn.LayerNorm.doctree deleted file mode 100644 index bdf459c54a..0000000000 Binary files a/docs/build/doctrees/python/nn/_autosummary/mlx.nn.LayerNorm.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/nn/_autosummary/mlx.nn.Linear.doctree b/docs/build/doctrees/python/nn/_autosummary/mlx.nn.Linear.doctree deleted file mode 100644 index b5945397d7..0000000000 Binary files a/docs/build/doctrees/python/nn/_autosummary/mlx.nn.Linear.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/nn/_autosummary/mlx.nn.Mish.doctree b/docs/build/doctrees/python/nn/_autosummary/mlx.nn.Mish.doctree deleted file mode 100644 index 96b277ed6b..0000000000 Binary files a/docs/build/doctrees/python/nn/_autosummary/mlx.nn.Mish.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/nn/_autosummary/mlx.nn.Module.apply.doctree b/docs/build/doctrees/python/nn/_autosummary/mlx.nn.Module.apply.doctree deleted file mode 100644 index 3ad46d0eed..0000000000 Binary files a/docs/build/doctrees/python/nn/_autosummary/mlx.nn.Module.apply.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/nn/_autosummary/mlx.nn.Module.apply_to_modules.doctree b/docs/build/doctrees/python/nn/_autosummary/mlx.nn.Module.apply_to_modules.doctree deleted file mode 100644 index f1f8d39147..0000000000 Binary files a/docs/build/doctrees/python/nn/_autosummary/mlx.nn.Module.apply_to_modules.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/nn/_autosummary/mlx.nn.Module.children.doctree b/docs/build/doctrees/python/nn/_autosummary/mlx.nn.Module.children.doctree deleted file mode 100644 index fda65019ec..0000000000 Binary files a/docs/build/doctrees/python/nn/_autosummary/mlx.nn.Module.children.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/nn/_autosummary/mlx.nn.Module.eval.doctree b/docs/build/doctrees/python/nn/_autosummary/mlx.nn.Module.eval.doctree deleted file mode 100644 index b70062947d..0000000000 Binary files a/docs/build/doctrees/python/nn/_autosummary/mlx.nn.Module.eval.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/nn/_autosummary/mlx.nn.Module.filter_and_map.doctree b/docs/build/doctrees/python/nn/_autosummary/mlx.nn.Module.filter_and_map.doctree deleted file mode 100644 index 875b776f8f..0000000000 Binary files a/docs/build/doctrees/python/nn/_autosummary/mlx.nn.Module.filter_and_map.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/nn/_autosummary/mlx.nn.Module.freeze.doctree b/docs/build/doctrees/python/nn/_autosummary/mlx.nn.Module.freeze.doctree deleted file mode 100644 index 9122645c83..0000000000 Binary files a/docs/build/doctrees/python/nn/_autosummary/mlx.nn.Module.freeze.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/nn/_autosummary/mlx.nn.Module.leaf_modules.doctree b/docs/build/doctrees/python/nn/_autosummary/mlx.nn.Module.leaf_modules.doctree deleted file mode 100644 index ef4a4b4922..0000000000 Binary files a/docs/build/doctrees/python/nn/_autosummary/mlx.nn.Module.leaf_modules.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/nn/_autosummary/mlx.nn.Module.load_weights.doctree b/docs/build/doctrees/python/nn/_autosummary/mlx.nn.Module.load_weights.doctree deleted file mode 100644 index 77dc55cbad..0000000000 Binary files a/docs/build/doctrees/python/nn/_autosummary/mlx.nn.Module.load_weights.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/nn/_autosummary/mlx.nn.Module.modules.doctree b/docs/build/doctrees/python/nn/_autosummary/mlx.nn.Module.modules.doctree deleted file mode 100644 index a8ed517227..0000000000 Binary files a/docs/build/doctrees/python/nn/_autosummary/mlx.nn.Module.modules.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/nn/_autosummary/mlx.nn.Module.named_modules.doctree b/docs/build/doctrees/python/nn/_autosummary/mlx.nn.Module.named_modules.doctree deleted file mode 100644 index 21d03145bc..0000000000 Binary files a/docs/build/doctrees/python/nn/_autosummary/mlx.nn.Module.named_modules.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/nn/_autosummary/mlx.nn.Module.parameters.doctree b/docs/build/doctrees/python/nn/_autosummary/mlx.nn.Module.parameters.doctree deleted file mode 100644 index bd44961317..0000000000 Binary files a/docs/build/doctrees/python/nn/_autosummary/mlx.nn.Module.parameters.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/nn/_autosummary/mlx.nn.Module.save_weights.doctree b/docs/build/doctrees/python/nn/_autosummary/mlx.nn.Module.save_weights.doctree deleted file mode 100644 index d8a9709a4e..0000000000 Binary files a/docs/build/doctrees/python/nn/_autosummary/mlx.nn.Module.save_weights.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/nn/_autosummary/mlx.nn.Module.train.doctree b/docs/build/doctrees/python/nn/_autosummary/mlx.nn.Module.train.doctree deleted file mode 100644 index c74bddc02a..0000000000 Binary files a/docs/build/doctrees/python/nn/_autosummary/mlx.nn.Module.train.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/nn/_autosummary/mlx.nn.Module.trainable_parameters.doctree b/docs/build/doctrees/python/nn/_autosummary/mlx.nn.Module.trainable_parameters.doctree deleted file mode 100644 index 9d931a5cfe..0000000000 Binary files a/docs/build/doctrees/python/nn/_autosummary/mlx.nn.Module.trainable_parameters.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/nn/_autosummary/mlx.nn.Module.training.doctree b/docs/build/doctrees/python/nn/_autosummary/mlx.nn.Module.training.doctree deleted file mode 100644 index 76044dd0ff..0000000000 Binary files a/docs/build/doctrees/python/nn/_autosummary/mlx.nn.Module.training.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/nn/_autosummary/mlx.nn.Module.unfreeze.doctree b/docs/build/doctrees/python/nn/_autosummary/mlx.nn.Module.unfreeze.doctree deleted file mode 100644 index 94b45be80e..0000000000 Binary files a/docs/build/doctrees/python/nn/_autosummary/mlx.nn.Module.unfreeze.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/nn/_autosummary/mlx.nn.Module.update.doctree b/docs/build/doctrees/python/nn/_autosummary/mlx.nn.Module.update.doctree deleted file mode 100644 index 322f33b28b..0000000000 Binary files a/docs/build/doctrees/python/nn/_autosummary/mlx.nn.Module.update.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/nn/_autosummary/mlx.nn.Module.update_modules.doctree b/docs/build/doctrees/python/nn/_autosummary/mlx.nn.Module.update_modules.doctree deleted file mode 100644 index 6c21b51863..0000000000 Binary files a/docs/build/doctrees/python/nn/_autosummary/mlx.nn.Module.update_modules.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/nn/_autosummary/mlx.nn.MultiHeadAttention.doctree b/docs/build/doctrees/python/nn/_autosummary/mlx.nn.MultiHeadAttention.doctree deleted file mode 100644 index bbe97731be..0000000000 Binary files a/docs/build/doctrees/python/nn/_autosummary/mlx.nn.MultiHeadAttention.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/nn/_autosummary/mlx.nn.PReLU.doctree b/docs/build/doctrees/python/nn/_autosummary/mlx.nn.PReLU.doctree deleted file mode 100644 index cf408b269f..0000000000 Binary files a/docs/build/doctrees/python/nn/_autosummary/mlx.nn.PReLU.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/nn/_autosummary/mlx.nn.QuantizedLinear.doctree b/docs/build/doctrees/python/nn/_autosummary/mlx.nn.QuantizedLinear.doctree deleted file mode 100644 index c29951241f..0000000000 Binary files a/docs/build/doctrees/python/nn/_autosummary/mlx.nn.QuantizedLinear.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/nn/_autosummary/mlx.nn.RMSNorm.doctree b/docs/build/doctrees/python/nn/_autosummary/mlx.nn.RMSNorm.doctree deleted file mode 100644 index 0ab225c966..0000000000 Binary files a/docs/build/doctrees/python/nn/_autosummary/mlx.nn.RMSNorm.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/nn/_autosummary/mlx.nn.ReLU.doctree b/docs/build/doctrees/python/nn/_autosummary/mlx.nn.ReLU.doctree deleted file mode 100644 index 2903dff65f..0000000000 Binary files a/docs/build/doctrees/python/nn/_autosummary/mlx.nn.ReLU.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/nn/_autosummary/mlx.nn.RoPE.doctree b/docs/build/doctrees/python/nn/_autosummary/mlx.nn.RoPE.doctree deleted file mode 100644 index 2772875e18..0000000000 Binary files a/docs/build/doctrees/python/nn/_autosummary/mlx.nn.RoPE.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/nn/_autosummary/mlx.nn.SELU.doctree b/docs/build/doctrees/python/nn/_autosummary/mlx.nn.SELU.doctree deleted file mode 100644 index 8a9a428628..0000000000 Binary files a/docs/build/doctrees/python/nn/_autosummary/mlx.nn.SELU.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/nn/_autosummary/mlx.nn.Sequential.doctree b/docs/build/doctrees/python/nn/_autosummary/mlx.nn.Sequential.doctree deleted file mode 100644 index bd76c52d83..0000000000 Binary files a/docs/build/doctrees/python/nn/_autosummary/mlx.nn.Sequential.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/nn/_autosummary/mlx.nn.SiLU.doctree b/docs/build/doctrees/python/nn/_autosummary/mlx.nn.SiLU.doctree deleted file mode 100644 index aa5817bab5..0000000000 Binary files a/docs/build/doctrees/python/nn/_autosummary/mlx.nn.SiLU.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/nn/_autosummary/mlx.nn.SinusoidalPositionalEncoding.doctree b/docs/build/doctrees/python/nn/_autosummary/mlx.nn.SinusoidalPositionalEncoding.doctree deleted file mode 100644 index 7840bf5a2a..0000000000 Binary files a/docs/build/doctrees/python/nn/_autosummary/mlx.nn.SinusoidalPositionalEncoding.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/nn/_autosummary/mlx.nn.Step.doctree b/docs/build/doctrees/python/nn/_autosummary/mlx.nn.Step.doctree deleted file mode 100644 index 87fff60721..0000000000 Binary files a/docs/build/doctrees/python/nn/_autosummary/mlx.nn.Step.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/nn/_autosummary/mlx.nn.Transformer.doctree b/docs/build/doctrees/python/nn/_autosummary/mlx.nn.Transformer.doctree deleted file mode 100644 index 87cde6d83d..0000000000 Binary files a/docs/build/doctrees/python/nn/_autosummary/mlx.nn.Transformer.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/nn/_autosummary_functions/mlx.nn.gelu.doctree b/docs/build/doctrees/python/nn/_autosummary_functions/mlx.nn.gelu.doctree deleted file mode 100644 index 92c3f6ee53..0000000000 Binary files a/docs/build/doctrees/python/nn/_autosummary_functions/mlx.nn.gelu.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/nn/_autosummary_functions/mlx.nn.gelu_approx.doctree b/docs/build/doctrees/python/nn/_autosummary_functions/mlx.nn.gelu_approx.doctree deleted file mode 100644 index 2bd6c87051..0000000000 Binary files a/docs/build/doctrees/python/nn/_autosummary_functions/mlx.nn.gelu_approx.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/nn/_autosummary_functions/mlx.nn.gelu_fast_approx.doctree b/docs/build/doctrees/python/nn/_autosummary_functions/mlx.nn.gelu_fast_approx.doctree deleted file mode 100644 index c714500d8f..0000000000 Binary files a/docs/build/doctrees/python/nn/_autosummary_functions/mlx.nn.gelu_fast_approx.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/nn/_autosummary_functions/mlx.nn.losses.binary_cross_entropy.doctree b/docs/build/doctrees/python/nn/_autosummary_functions/mlx.nn.losses.binary_cross_entropy.doctree deleted file mode 100644 index 108e75c4f7..0000000000 Binary files a/docs/build/doctrees/python/nn/_autosummary_functions/mlx.nn.losses.binary_cross_entropy.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/nn/_autosummary_functions/mlx.nn.losses.cosine_similarity_loss.doctree b/docs/build/doctrees/python/nn/_autosummary_functions/mlx.nn.losses.cosine_similarity_loss.doctree deleted file mode 100644 index f8126e969b..0000000000 Binary files a/docs/build/doctrees/python/nn/_autosummary_functions/mlx.nn.losses.cosine_similarity_loss.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/nn/_autosummary_functions/mlx.nn.losses.cross_entropy.doctree b/docs/build/doctrees/python/nn/_autosummary_functions/mlx.nn.losses.cross_entropy.doctree deleted file mode 100644 index 9cdd9958b2..0000000000 Binary files a/docs/build/doctrees/python/nn/_autosummary_functions/mlx.nn.losses.cross_entropy.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/nn/_autosummary_functions/mlx.nn.losses.hinge_loss.doctree b/docs/build/doctrees/python/nn/_autosummary_functions/mlx.nn.losses.hinge_loss.doctree deleted file mode 100644 index 9014d65307..0000000000 Binary files a/docs/build/doctrees/python/nn/_autosummary_functions/mlx.nn.losses.hinge_loss.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/nn/_autosummary_functions/mlx.nn.losses.huber_loss.doctree b/docs/build/doctrees/python/nn/_autosummary_functions/mlx.nn.losses.huber_loss.doctree deleted file mode 100644 index c040f2d7f5..0000000000 Binary files a/docs/build/doctrees/python/nn/_autosummary_functions/mlx.nn.losses.huber_loss.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/nn/_autosummary_functions/mlx.nn.losses.kl_div_loss.doctree b/docs/build/doctrees/python/nn/_autosummary_functions/mlx.nn.losses.kl_div_loss.doctree deleted file mode 100644 index dfbd521e73..0000000000 Binary files a/docs/build/doctrees/python/nn/_autosummary_functions/mlx.nn.losses.kl_div_loss.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/nn/_autosummary_functions/mlx.nn.losses.l1_loss.doctree b/docs/build/doctrees/python/nn/_autosummary_functions/mlx.nn.losses.l1_loss.doctree deleted file mode 100644 index 84495517a3..0000000000 Binary files a/docs/build/doctrees/python/nn/_autosummary_functions/mlx.nn.losses.l1_loss.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/nn/_autosummary_functions/mlx.nn.losses.log_cosh_loss.doctree b/docs/build/doctrees/python/nn/_autosummary_functions/mlx.nn.losses.log_cosh_loss.doctree deleted file mode 100644 index 718c7c5e40..0000000000 Binary files a/docs/build/doctrees/python/nn/_autosummary_functions/mlx.nn.losses.log_cosh_loss.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/nn/_autosummary_functions/mlx.nn.losses.mse_loss.doctree b/docs/build/doctrees/python/nn/_autosummary_functions/mlx.nn.losses.mse_loss.doctree deleted file mode 100644 index a34c3b1551..0000000000 Binary files a/docs/build/doctrees/python/nn/_autosummary_functions/mlx.nn.losses.mse_loss.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/nn/_autosummary_functions/mlx.nn.losses.nll_loss.doctree b/docs/build/doctrees/python/nn/_autosummary_functions/mlx.nn.losses.nll_loss.doctree deleted file mode 100644 index 0d95d52a90..0000000000 Binary files a/docs/build/doctrees/python/nn/_autosummary_functions/mlx.nn.losses.nll_loss.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/nn/_autosummary_functions/mlx.nn.losses.smooth_l1_loss.doctree b/docs/build/doctrees/python/nn/_autosummary_functions/mlx.nn.losses.smooth_l1_loss.doctree deleted file mode 100644 index b1965fa8fd..0000000000 Binary files a/docs/build/doctrees/python/nn/_autosummary_functions/mlx.nn.losses.smooth_l1_loss.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/nn/_autosummary_functions/mlx.nn.losses.triplet_loss.doctree b/docs/build/doctrees/python/nn/_autosummary_functions/mlx.nn.losses.triplet_loss.doctree deleted file mode 100644 index 7268329895..0000000000 Binary files a/docs/build/doctrees/python/nn/_autosummary_functions/mlx.nn.losses.triplet_loss.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/nn/_autosummary_functions/mlx.nn.mish.doctree b/docs/build/doctrees/python/nn/_autosummary_functions/mlx.nn.mish.doctree deleted file mode 100644 index 2e3c4d245b..0000000000 Binary files a/docs/build/doctrees/python/nn/_autosummary_functions/mlx.nn.mish.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/nn/_autosummary_functions/mlx.nn.prelu.doctree b/docs/build/doctrees/python/nn/_autosummary_functions/mlx.nn.prelu.doctree deleted file mode 100644 index 0c97a551e3..0000000000 Binary files a/docs/build/doctrees/python/nn/_autosummary_functions/mlx.nn.prelu.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/nn/_autosummary_functions/mlx.nn.relu.doctree b/docs/build/doctrees/python/nn/_autosummary_functions/mlx.nn.relu.doctree deleted file mode 100644 index 398e9179f1..0000000000 Binary files a/docs/build/doctrees/python/nn/_autosummary_functions/mlx.nn.relu.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/nn/_autosummary_functions/mlx.nn.selu.doctree b/docs/build/doctrees/python/nn/_autosummary_functions/mlx.nn.selu.doctree deleted file mode 100644 index 3e4da6c661..0000000000 Binary files a/docs/build/doctrees/python/nn/_autosummary_functions/mlx.nn.selu.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/nn/_autosummary_functions/mlx.nn.silu.doctree b/docs/build/doctrees/python/nn/_autosummary_functions/mlx.nn.silu.doctree deleted file mode 100644 index 62bde63c8e..0000000000 Binary files a/docs/build/doctrees/python/nn/_autosummary_functions/mlx.nn.silu.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/nn/_autosummary_functions/mlx.nn.step.doctree b/docs/build/doctrees/python/nn/_autosummary_functions/mlx.nn.step.doctree deleted file mode 100644 index 46c14e3e5f..0000000000 Binary files a/docs/build/doctrees/python/nn/_autosummary_functions/mlx.nn.step.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/nn/functions.doctree b/docs/build/doctrees/python/nn/functions.doctree deleted file mode 100644 index 5d54c1c5da..0000000000 Binary files a/docs/build/doctrees/python/nn/functions.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/nn/layers.doctree b/docs/build/doctrees/python/nn/layers.doctree deleted file mode 100644 index 6eaf2aa685..0000000000 Binary files a/docs/build/doctrees/python/nn/layers.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/nn/losses.doctree b/docs/build/doctrees/python/nn/losses.doctree deleted file mode 100644 index 52f541c36d..0000000000 Binary files a/docs/build/doctrees/python/nn/losses.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/nn/module.doctree b/docs/build/doctrees/python/nn/module.doctree deleted file mode 100644 index a7c439af29..0000000000 Binary files a/docs/build/doctrees/python/nn/module.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/ops.doctree b/docs/build/doctrees/python/ops.doctree deleted file mode 100644 index 967eee39b0..0000000000 Binary files a/docs/build/doctrees/python/ops.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/optimizers.doctree b/docs/build/doctrees/python/optimizers.doctree deleted file mode 100644 index 81dd2455da..0000000000 Binary files a/docs/build/doctrees/python/optimizers.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/random.doctree b/docs/build/doctrees/python/random.doctree deleted file mode 100644 index 8cf3f5aa8b..0000000000 Binary files a/docs/build/doctrees/python/random.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/transforms.doctree b/docs/build/doctrees/python/transforms.doctree deleted file mode 100644 index 03a963292c..0000000000 Binary files a/docs/build/doctrees/python/transforms.doctree and /dev/null differ diff --git a/docs/build/doctrees/python/tree_utils.doctree b/docs/build/doctrees/python/tree_utils.doctree deleted file mode 100644 index 0dce281837..0000000000 Binary files a/docs/build/doctrees/python/tree_utils.doctree and /dev/null differ diff --git a/docs/build/doctrees/usage/function_transforms.doctree b/docs/build/doctrees/usage/function_transforms.doctree deleted file mode 100644 index 43a605bb42..0000000000 Binary files a/docs/build/doctrees/usage/function_transforms.doctree and /dev/null differ diff --git a/docs/build/doctrees/usage/indexing.doctree b/docs/build/doctrees/usage/indexing.doctree deleted file mode 100644 index 1057dbc09b..0000000000 Binary files a/docs/build/doctrees/usage/indexing.doctree and /dev/null differ diff --git a/docs/build/doctrees/usage/lazy_evaluation.doctree b/docs/build/doctrees/usage/lazy_evaluation.doctree deleted file mode 100644 index 246d9817d6..0000000000 Binary files a/docs/build/doctrees/usage/lazy_evaluation.doctree and /dev/null differ diff --git a/docs/build/doctrees/usage/numpy.doctree b/docs/build/doctrees/usage/numpy.doctree deleted file mode 100644 index 0270c9da06..0000000000 Binary files a/docs/build/doctrees/usage/numpy.doctree and /dev/null differ diff --git a/docs/build/doctrees/usage/quick_start.doctree b/docs/build/doctrees/usage/quick_start.doctree deleted file mode 100644 index 9788c75bfa..0000000000 Binary files a/docs/build/doctrees/usage/quick_start.doctree and /dev/null differ diff --git a/docs/build/doctrees/usage/saving_and_loading.doctree b/docs/build/doctrees/usage/saving_and_loading.doctree deleted file mode 100644 index ce6f41b860..0000000000 Binary files a/docs/build/doctrees/usage/saving_and_loading.doctree and /dev/null differ diff --git a/docs/build/doctrees/usage/unified_memory.doctree b/docs/build/doctrees/usage/unified_memory.doctree deleted file mode 100644 index f4e516386b..0000000000 Binary files a/docs/build/doctrees/usage/unified_memory.doctree and /dev/null differ diff --git a/docs/build/doctrees/usage/using_streams.doctree b/docs/build/doctrees/usage/using_streams.doctree deleted file mode 100644 index af6d4f6b93..0000000000 Binary files a/docs/build/doctrees/usage/using_streams.doctree and /dev/null differ