docs update

docs/build/html/_sources/python/_autosummary/mlx.core.arctan2.rst

@@ -0,0 +1,6 @@
+mlx.core.arctan2
+================
+
+.. currentmodule:: mlx.core
+
+.. autofunction:: arctan2

docs/build/html/_sources/python/_autosummary/mlx.core.array.rst

@@ -20,6 +20,7 @@
       ~array.argmax
       ~array.argmin
       ~array.astype
+      ~array.conj
       ~array.cos
       ~array.cummax
       ~array.cummin

docs/build/html/_sources/python/_autosummary/mlx.core.bitwise_and.rst

@@ -0,0 +1,6 @@
+mlx.core.bitwise\_and
+=====================
+
+.. currentmodule:: mlx.core
+
+.. autofunction:: bitwise_and

docs/build/html/_sources/python/_autosummary/mlx.core.bitwise_or.rst

@@ -0,0 +1,6 @@
+mlx.core.bitwise\_or
+====================
+
+.. currentmodule:: mlx.core
+
+.. autofunction:: bitwise_or

docs/build/html/_sources/python/_autosummary/mlx.core.bitwise_xor.rst

@@ -0,0 +1,6 @@
+mlx.core.bitwise\_xor
+=====================
+
+.. currentmodule:: mlx.core
+
+.. autofunction:: bitwise_xor

docs/build/html/_sources/python/_autosummary/mlx.core.block_sparse_mm.rst

@@ -0,0 +1,6 @@
+mlx.core.block\_sparse\_mm
+==========================
+
+.. currentmodule:: mlx.core
+
+.. autofunction:: block_sparse_mm

docs/build/html/_sources/python/_autosummary/mlx.core.conj.rst

@@ -0,0 +1,6 @@
+mlx.core.conj
+=============
+
+.. currentmodule:: mlx.core
+
+.. autofunction:: conj

docs/build/html/_sources/python/_autosummary/mlx.core.conjugate.rst

@@ -0,0 +1,6 @@
+mlx.core.conjugate
+==================
+
+.. currentmodule:: mlx.core
+
+.. autofunction:: conjugate

docs/build/html/_sources/python/_autosummary/mlx.core.left_shift.rst

@@ -0,0 +1,6 @@
+mlx.core.left\_shift
+====================
+
+.. currentmodule:: mlx.core
+
+.. autofunction:: left_shift

docs/build/html/_sources/python/_autosummary/mlx.core.metal.device_info.rst

@@ -0,0 +1,6 @@
+mlx.core.metal.device\_info
+===========================
+
+.. currentmodule:: mlx.core.metal
+
+.. autofunction:: device_info

docs/build/html/_sources/python/_autosummary/mlx.core.metal.reset_peak_memory.rst

@@ -0,0 +1,6 @@
+mlx.core.metal.reset\_peak\_memory
+==================================
+
+.. currentmodule:: mlx.core.metal
+
+.. autofunction:: reset_peak_memory

docs/build/html/_sources/python/_autosummary/mlx.core.right_shift.rst

@@ -0,0 +1,6 @@
+mlx.core.right\_shift
+=====================
+
+.. currentmodule:: mlx.core
+
+.. autofunction:: right_shift

docs/build/html/_sources/python/_autosummary/mlx.optimizers.clip_grad_norm.rst

@@ -0,0 +1,6 @@
+mlx.optimizers.clip\_grad\_norm
+===============================
+
+.. currentmodule:: mlx.optimizers
+
+.. autofunction:: clip_grad_norm

docs/build/html/_sources/python/_autosummary/mlx.utils.tree_reduce.rst

@@ -0,0 +1,6 @@
+mlx.utils.tree\_reduce
+======================
+
+.. currentmodule:: mlx.utils
+
+.. autofunction:: tree_reduce

docs/build/html/_sources/python/metal.rst

@@ -7,8 +7,10 @@ Metal
   :toctree: _autosummary
 
   is_available
+  device_info
   get_active_memory
   get_peak_memory
+  reset_peak_memory
   get_cache_memory
   set_memory_limit
   set_cache_limit

docs/build/html/_sources/python/ops.rst

@@ -19,6 +19,7 @@ Operations
    arcsin
    arcsinh
    arctan
+   arctan2
    arctanh
    argmax
    argmin
@@ -28,11 +29,17 @@ Operations
    atleast_1d
    atleast_2d
    atleast_3d
-   broadcast_to
+   bitwise_and
+   bitwise_or
+   bitwise_xor
    block_masked_mm
+   block_sparse_mm
+   broadcast_to
    ceil
    clip
    concatenate
+   conj
+   conjugate
    convolve
    conv1d
    conv2d
@@ -69,6 +76,7 @@ Operations
    isnan
    isneginf
    isposinf
+   left_shift
    less
    less_equal
    linspace
@@ -105,6 +113,7 @@ Operations
    reciprocal
    repeat
    reshape
+   right_shift
    round
    rsqrt
    save

docs/build/html/_sources/python/optimizers.rst

@@ -1,5 +1,7 @@
 .. _optimizers:
 
+.. currentmodule:: mlx.optimizers
+
 Optimizers
 ==========
 
@@ -34,3 +36,8 @@ model's parameters and the **optimizer state**.
    optimizers/optimizer
    optimizers/common_optimizers
    optimizers/schedulers
+
+.. autosummary::
+   :toctree: _autosummary
+
+   clip_grad_norm

docs/build/html/_sources/python/tree_utils.rst

@@ -20,3 +20,4 @@ return python trees will be using the default python ``dict``, ``list`` and
    tree_unflatten
    tree_map
    tree_map_with_path
+   tree_reduce