diff --git a/docs/build/doctrees/cpp/ops.doctree b/docs/build/doctrees/cpp/ops.doctree
deleted file mode 100644
index 8ade67022..000000000
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 a0d2baf1d..000000000
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 2f7d96493..000000000
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 4b9be359e..000000000
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 23804ca7a..000000000
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 20decac88..000000000
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 96bbda118..000000000
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 8d5f88809..000000000
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 0e8b53837..000000000
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 b76aa6100..000000000
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 b09770dda..000000000
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 8d6d05722..000000000
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 f2286c5aa..000000000
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 b8e2f1d37..000000000
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 657f0c24b..000000000
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 b7da1ff32..000000000
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 0fd022e89..000000000
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 f7ab0f64a..000000000
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 e251f756c..000000000
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 4eaeb9945..000000000
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 d567a4b93..000000000
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 7e1681e42..000000000
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 4203ad352..000000000
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 7f3ef214f..000000000
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 a85e59e58..000000000
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 a309dea13..000000000
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 17aaf5d4a..000000000
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 8a845dddc..000000000
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 942979806..000000000
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 79f01e39d..000000000
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 98eb5675f..000000000
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 78d192cbe..000000000
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 ea2043861..000000000
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 1dd1944a7..000000000
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 5903c60b3..000000000
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 43418f8c8..000000000
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 19bef20a7..000000000
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 6a32c5103..000000000
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 ec88b9d46..000000000
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 db73a238c..000000000
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 3840dcd76..000000000
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 1ecb477ab..000000000
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 261fdf7ba..000000000
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 c14cb4fb0..000000000
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 9a739fba5..000000000
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 12ff80192..000000000
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 643d5471c..000000000
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 3156e2020..000000000
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 f52a86511..000000000
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 9afc4c91b..000000000
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 3c275626b..000000000
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 2f6183431..000000000
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 51752c2a9..000000000
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 c73cad869..000000000
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 8a1f5e9e7..000000000
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 0045103e0..000000000
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 9ed6d6f8b..000000000
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 d8a5adefa..000000000
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 d7866eb13..000000000
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 f7922948f..000000000
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 1067c4261..000000000
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 77eec964b..000000000
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 6dca61080..000000000
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 f55398aa2..000000000
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 bcf04a497..000000000
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 cdbf97110..000000000
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 77b1b8ca2..000000000
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 9eacea9e0..000000000
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 c4823f85c..000000000
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 035779f02..000000000
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 474dfbf72..000000000
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 f530d1da5..000000000
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 7198c13a9..000000000
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 1e05a407d..000000000
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 964b2400c..000000000
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 dfdfaca64..000000000
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 cd39a49c6..000000000
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 ddc2eefed..000000000
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 1676d7dbc..000000000
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 e9a0ba367..000000000
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 a5515bc24..000000000
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 2b4cc7a1c..000000000
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 5f93ca3e2..000000000
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 279ef6473..000000000
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 406878c3c..000000000
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 06fe68acc..000000000
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 c23ebb2cb..000000000
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 3bc531ea8..000000000
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 640ca7896..000000000
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 8372bf45e..000000000
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 6cb1aaddf..000000000
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 253424bd4..000000000
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 87e93b544..000000000
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 1c61d4992..000000000
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 3c41914bf..000000000
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 841f020bd..000000000
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 94cfb529b..000000000
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 8d63347ec..000000000
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 ca8e07c20..000000000
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 bd99443dc..000000000
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 0e625e342..000000000
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 5573eb556..000000000
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 e11d9bd2d..000000000
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 d48aa6bc8..000000000
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 830ebb9a4..000000000
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 8322e6451..000000000
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 3cb9d80d6..000000000
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 aac5c0a6a..000000000
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 7c6fbd1b1..000000000
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 f9e957686..000000000
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 808bf41e6..000000000
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 4bf27e832..000000000
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 03abea113..000000000
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 eee3858aa..000000000
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 0b4027e64..000000000
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 28848956a..000000000
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 91a06483e..000000000
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 62d2e0dad..000000000
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 3b7cc020d..000000000
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 9dfc94bfd..000000000
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 43109d0cb..000000000
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 b6db9d22a..000000000
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 2ecb37256..000000000
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 31c0b2bd4..000000000
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 c13fe054f..000000000
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 ef48dff78..000000000
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 effa5331b..000000000
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 78741b97a..000000000
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 8d836911f..000000000
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 1c39b81ae..000000000
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 c374fa7e7..000000000
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 13feb8655..000000000
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 e9cf8d29e..000000000
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 293e89f6c..000000000
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 5576b0db8..000000000
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 43ac2a1d0..000000000
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 de3a605fd..000000000
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 e70afcebc..000000000
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 baf59662e..000000000
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 cd5d60f56..000000000
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 28716d48a..000000000
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 51bd41931..000000000
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 9cf4871c4..000000000
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 b49ceb730..000000000
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 7c388509d..000000000
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 c59a5f8aa..000000000
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 c66cc8b90..000000000
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 a8f54cc52..000000000
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 fdbc46917..000000000
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 2319da031..000000000
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 61200ae17..000000000
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 7151f851a..000000000
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 f65c41fa8..000000000
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 f3f77d856..000000000
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 21f35145b..000000000
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 26bb22da0..000000000
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 62c6b196c..000000000
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 e5603bc97..000000000
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 fde220212..000000000
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 5bb76b6dc..000000000
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 2363ee48a..000000000
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 5484b3d35..000000000
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 3b72c0325..000000000
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 4acb22b29..000000000
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 1a099b4ee..000000000
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 c8d8e4a8e..000000000
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 de8e10ad5..000000000
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 acf0171e2..000000000
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 5a5a74b3c..000000000
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 d6aa3f72e..000000000
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 ee68048f0..000000000
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 3c8c33a8b..000000000
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 602fbe023..000000000
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 9b8a960b9..000000000
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 96b5b740e..000000000
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 2f5d6ba12..000000000
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 295b0f412..000000000
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 2997e4271..000000000
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 dcde15d9f..000000000
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 8512009d9..000000000
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 b1644987b..000000000
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 fabce8aff..000000000
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 38b040c87..000000000
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 811252803..000000000
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 994e06fcc..000000000
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 c8e4f3593..000000000
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 8920df3fb..000000000
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 52fabce5d..000000000
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 c6a282a28..000000000
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 0513ecacc..000000000
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 f89559aba..000000000
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 868898ddb..000000000
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 9cb374060..000000000
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 e4499ae4c..000000000
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 3682d9770..000000000
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 96a0ef226..000000000
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 65030721a..000000000
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 4b414542a..000000000
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 e3291a8d1..000000000
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 cfe6a6c0d..000000000
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 5bbe3c40b..000000000
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 4fa8ddcc0..000000000
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 921aca155..000000000
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 d4a69be2e..000000000
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 8f8379715..000000000
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 215c29f80..000000000
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 cab935487..000000000
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 e7351cc08..000000000
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 703f7e65f..000000000
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 e36586447..000000000
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 d4ad20ae3..000000000
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 6a30a134c..000000000
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 2bede7452..000000000
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 87a09a6a9..000000000
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 93ceef54f..000000000
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 ce1b0d928..000000000
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 556c5024f..000000000
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 2f1d10293..000000000
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 779bcd026..000000000
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 bc3c9ccc3..000000000
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 9f172fb41..000000000
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 60a8d9a2f..000000000
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 684a41cde..000000000
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 2304bd122..000000000
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 9e7d9afd2..000000000
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 a6f01f388..000000000
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 bdf459c54..000000000
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 b5945397d..000000000
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 96b277ed6..000000000
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 3ad46d0ee..000000000
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 f1f8d3914..000000000
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 fda65019e..000000000
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 b70062947..000000000
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 875b776f8..000000000
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 9122645c8..000000000
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 ef4a4b492..000000000
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 77dc55cba..000000000
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 a8ed51722..000000000
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 21d03145b..000000000
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 bd4496131..000000000
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 d8a9709a4..000000000
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 c74bddc02..000000000
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 9d931a5cf..000000000
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 76044dd0f..000000000
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 94b45be80..000000000
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 322f33b28..000000000
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 6c21b5186..000000000
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 bbe97731b..000000000
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 cf408b269..000000000
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 c29951241..000000000
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 0ab225c96..000000000
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 2903dff65..000000000
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 2772875e1..000000000
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 8a9a42862..000000000
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 bd76c52d8..000000000
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 aa5817bab..000000000
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 7840bf5a2..000000000
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 87fff6072..000000000
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 87cde6d83..000000000
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 92c3f6ee5..000000000
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 2bd6c8705..000000000
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 c714500d8..000000000
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 108e75c4f..000000000
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 f8126e969..000000000
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 9cdd9958b..000000000
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 9014d6530..000000000
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 c040f2d7f..000000000
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 dfbd521e7..000000000
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 84495517a..000000000
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 718c7c5e4..000000000
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 a34c3b155..000000000
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 0d95d52a9..000000000
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 b1965fa8f..000000000
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 726832989..000000000
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 2e3c4d245..000000000
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 0c97a551e..000000000
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 398e9179f..000000000
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 3e4da6c66..000000000
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 62bde63c8..000000000
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 46c14e3e5..000000000
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 5d54c1c5d..000000000
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 6eaf2aa68..000000000
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 52f541c36..000000000
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 a7c439af2..000000000
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 967eee39b..000000000
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 81dd2455d..000000000
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 8cf3f5aa8..000000000
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 03a963292..000000000
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 0dce28183..000000000
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 43a605bb4..000000000
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 1057dbc09..000000000
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 246d9817d..000000000
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 0270c9da0..000000000
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 9788c75bf..000000000
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 ce6f41b86..000000000
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 f4e516386..000000000
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 af6d4f6b9..000000000
Binary files a/docs/build/doctrees/usage/using_streams.doctree and /dev/null differ