docs update

This commit is contained in:
Awni Hannun
2024-06-06 20:28:06 -07:00
committed by CircleCI Docs
parent 2360620475
commit ba4eff9520
2046 changed files with 323202 additions and 2319 deletions

View File

@@ -55,6 +55,7 @@
~array.tolist
~array.transpose
~array.var
~array.view

View File

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

View File

@@ -0,0 +1,25 @@
mlx.core.distributed.Group
==========================
.. currentmodule:: mlx.core.distributed
.. autoclass:: Group
.. automethod:: __init__
.. rubric:: Methods
.. autosummary::
~Group.__init__
~Group.rank
~Group.size
~Group.split

View File

@@ -0,0 +1,6 @@
mlx.core.distributed.all\_gather
================================
.. currentmodule:: mlx.core.distributed
.. autofunction:: all_gather

View File

@@ -0,0 +1,6 @@
mlx.core.distributed.all\_sum
=============================
.. currentmodule:: mlx.core.distributed
.. autofunction:: all_sum

View File

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

View File

@@ -0,0 +1,6 @@
mlx.core.distributed.is\_available
==================================
.. currentmodule:: mlx.core.distributed
.. autofunction:: is_available

View File

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