diff --git a/docs/build/doctrees/cpp/ops.doctree b/docs/build/doctrees/cpp/ops.doctree
deleted file mode 100644
index 8ade6702..00000000
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 a0d2baf1..00000000
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 2f7d9649..00000000
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 4b9be359..00000000
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 23804ca7..00000000
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 20decac8..00000000
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 96bbda11..00000000
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 8d5f8880..00000000
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 0e8b5383..00000000
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 b76aa610..00000000
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 b09770dd..00000000
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 8d6d0572..00000000
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 f2286c5a..00000000
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 b8e2f1d3..00000000
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 657f0c24..00000000
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 b7da1ff3..00000000
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 0fd022e8..00000000
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 f7ab0f64..00000000
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 e251f756..00000000
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 4eaeb994..00000000
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 d567a4b9..00000000
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 7e1681e4..00000000
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 4203ad35..00000000
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 7f3ef214..00000000
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 a85e59e5..00000000
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 a309dea1..00000000
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 17aaf5d4..00000000
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 8a845ddd..00000000
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 94297980..00000000
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 79f01e39..00000000
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 98eb5675..00000000
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 78d192cb..00000000
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 ea204386..00000000
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 1dd1944a..00000000
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 5903c60b..00000000
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 43418f8c..00000000
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 19bef20a..00000000
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 6a32c510..00000000
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 ec88b9d4..00000000
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 db73a238..00000000
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 3840dcd7..00000000
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 1ecb477a..00000000
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 261fdf7b..00000000
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 c14cb4fb..00000000
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 9a739fba..00000000
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 12ff8019..00000000
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 643d5471..00000000
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 3156e202..00000000
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 f52a8651..00000000
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 9afc4c91..00000000
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 3c275626..00000000
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 2f618343..00000000
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 51752c2a..00000000
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 c73cad86..00000000
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 8a1f5e9e..00000000
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 0045103e..00000000
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 9ed6d6f8..00000000
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 d8a5adef..00000000
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 d7866eb1..00000000
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 f7922948..00000000
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 1067c426..00000000
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 77eec964..00000000
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 6dca6108..00000000
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 f55398aa..00000000
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 bcf04a49..00000000
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 cdbf9711..00000000
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 77b1b8ca..00000000
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 9eacea9e..00000000
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 c4823f85..00000000
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 035779f0..00000000
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 474dfbf7..00000000
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 f530d1da..00000000
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 7198c13a..00000000
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 1e05a407..00000000
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 964b2400..00000000
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 dfdfaca6..00000000
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 cd39a49c..00000000
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 ddc2eefe..00000000
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 1676d7db..00000000
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 e9a0ba36..00000000
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 a5515bc2..00000000
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 2b4cc7a1..00000000
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 5f93ca3e..00000000
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 279ef647..00000000
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 406878c3..00000000
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 06fe68ac..00000000
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 c23ebb2c..00000000
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 3bc531ea..00000000
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 640ca789..00000000
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 8372bf45..00000000
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 6cb1aadd..00000000
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 253424bd..00000000
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 87e93b54..00000000
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 1c61d499..00000000
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 3c41914b..00000000
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 841f020b..00000000
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 94cfb529..00000000
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 8d63347e..00000000
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 ca8e07c2..00000000
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 bd99443d..00000000
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 0e625e34..00000000
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 5573eb55..00000000
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 e11d9bd2..00000000
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 d48aa6bc..00000000
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 830ebb9a..00000000
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 8322e645..00000000
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 3cb9d80d..00000000
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 aac5c0a6..00000000
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 7c6fbd1b..00000000
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 f9e95768..00000000
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 808bf41e..00000000
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 4bf27e83..00000000
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 03abea11..00000000
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 eee3858a..00000000
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 0b4027e6..00000000
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 28848956..00000000
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 91a06483..00000000
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 62d2e0da..00000000
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 3b7cc020..00000000
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 9dfc94bf..00000000
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 43109d0c..00000000
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 b6db9d22..00000000
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 2ecb3725..00000000
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 31c0b2bd..00000000
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 c13fe054..00000000
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 ef48dff7..00000000
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 effa5331..00000000
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 78741b97..00000000
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 8d836911..00000000
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 1c39b81a..00000000
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 c374fa7e..00000000
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 13feb865..00000000
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 e9cf8d29..00000000
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 293e89f6..00000000
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 5576b0db..00000000
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 43ac2a1d..00000000
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 de3a605f..00000000
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 e70afceb..00000000
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 baf59662..00000000
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 cd5d60f5..00000000
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 28716d48..00000000
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 51bd4193..00000000
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 9cf4871c..00000000
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 b49ceb73..00000000
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 7c388509..00000000
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 c59a5f8a..00000000
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 c66cc8b9..00000000
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 a8f54cc5..00000000
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 fdbc4691..00000000
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 2319da03..00000000
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 61200ae1..00000000
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 7151f851..00000000
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 f65c41fa..00000000
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 f3f77d85..00000000
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 21f35145..00000000
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 26bb22da..00000000
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 62c6b196..00000000
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 e5603bc9..00000000
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 fde22021..00000000
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 5bb76b6d..00000000
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 2363ee48..00000000
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 5484b3d3..00000000
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 3b72c032..00000000
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 4acb22b2..00000000
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 1a099b4e..00000000
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 c8d8e4a8..00000000
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 de8e10ad..00000000
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 acf0171e..00000000
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 5a5a74b3..00000000
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 d6aa3f72..00000000
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 ee68048f..00000000
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 3c8c33a8..00000000
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 602fbe02..00000000
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 9b8a960b..00000000
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 96b5b740..00000000
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 2f5d6ba1..00000000
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 295b0f41..00000000
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 2997e427..00000000
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 dcde15d9..00000000
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 8512009d..00000000
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 b1644987..00000000
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 fabce8af..00000000
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 38b040c8..00000000
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 81125280..00000000
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 994e06fc..00000000
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 c8e4f359..00000000
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 8920df3f..00000000
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 52fabce5..00000000
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 c6a282a2..00000000
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 0513ecac..00000000
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 f89559ab..00000000
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 868898dd..00000000
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 9cb37406..00000000
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 e4499ae4..00000000
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 3682d977..00000000
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 96a0ef22..00000000
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 65030721..00000000
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 4b414542..00000000
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 e3291a8d..00000000
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 cfe6a6c0..00000000
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 5bbe3c40..00000000
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 4fa8ddcc..00000000
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 921aca15..00000000
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 d4a69be2..00000000
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 8f837971..00000000
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 215c29f8..00000000
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 cab93548..00000000
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 e7351cc0..00000000
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 703f7e65..00000000
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 e3658644..00000000
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 d4ad20ae..00000000
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 6a30a134..00000000
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 2bede745..00000000
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 87a09a6a..00000000
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 93ceef54..00000000
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 ce1b0d92..00000000
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 556c5024..00000000
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 2f1d1029..00000000
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 779bcd02..00000000
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 bc3c9ccc..00000000
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 9f172fb4..00000000
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 60a8d9a2..00000000
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 684a41cd..00000000
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 2304bd12..00000000
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 9e7d9afd..00000000
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 a6f01f38..00000000
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 bdf459c5..00000000
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 b5945397..00000000
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 96b277ed..00000000
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 3ad46d0e..00000000
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 f1f8d391..00000000
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 fda65019..00000000
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 b7006294..00000000
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 875b776f..00000000
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 9122645c..00000000
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 ef4a4b49..00000000
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 77dc55cb..00000000
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 a8ed5172..00000000
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 21d03145..00000000
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 bd449613..00000000
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 d8a9709a..00000000
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 c74bddc0..00000000
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 9d931a5c..00000000
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 76044dd0..00000000
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 94b45be8..00000000
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 322f33b2..00000000
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 6c21b518..00000000
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 bbe97731..00000000
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 cf408b26..00000000
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 c2995124..00000000
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 0ab225c9..00000000
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 2903dff6..00000000
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 2772875e..00000000
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 8a9a4286..00000000
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 bd76c52d..00000000
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 aa5817ba..00000000
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 7840bf5a..00000000
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 87fff607..00000000
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 87cde6d8..00000000
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 92c3f6ee..00000000
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 2bd6c870..00000000
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 c714500d..00000000
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 108e75c4..00000000
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 f8126e96..00000000
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 9cdd9958..00000000
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 9014d653..00000000
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 c040f2d7..00000000
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 dfbd521e..00000000
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 84495517..00000000
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 718c7c5e..00000000
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 a34c3b15..00000000
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 0d95d52a..00000000
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 b1965fa8..00000000
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 72683298..00000000
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 2e3c4d24..00000000
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 0c97a551..00000000
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 398e9179..00000000
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 3e4da6c6..00000000
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 62bde63c..00000000
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 46c14e3e..00000000
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 5d54c1c5..00000000
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 6eaf2aa6..00000000
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 52f541c3..00000000
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 a7c439af..00000000
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 967eee39..00000000
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 81dd2455..00000000
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 8cf3f5aa..00000000
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 03a96329..00000000
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 0dce2818..00000000
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 43a605bb..00000000
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 1057dbc0..00000000
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 246d9817..00000000
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 0270c9da..00000000
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 9788c75b..00000000
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 ce6f41b8..00000000
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 f4e51638..00000000
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 af6d4f6b..00000000
Binary files a/docs/build/doctrees/usage/using_streams.doctree and /dev/null differ