zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-06-24 17:31:16 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

apple mlr (#7)

Browse Source
This commit is contained in:
Awni Hannun 2023-12-05 14:10:59 -08:00 committed by GitHub
parent 6449a8682a
commit 49cda449b1
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 5 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
README.md
View File

@ -4,7 +4,8 @@
[**Documentation**](https://ml-explore.github.io/mlx/build/html/index.html) | [**Documentation**](https://ml-explore.github.io/mlx/build/html/index.html) |
[**Examples**](#examples) [**Examples**](#examples)
MLX is an array framework for machine learning on Apple silicon. MLX is an array framework for machine learning on Apple silicon, brought to you
by Apple machine learning research.
Some key features of MLX include: Some key features of MLX include:

6
docs/src/index.rst
View File

@ -1,8 +1,8 @@
MLX MLX
=== ===
MLX is a NumPy-like array framework designed for efficient and flexible MLX is a NumPy-like array framework designed for efficient and flexible machine
machine learning on Apple silicon. learning on Apple silicon, brought to you by Apple machine learning research.
The Python API closely follows NumPy with a few exceptions. MLX also has a The Python API closely follows NumPy with a few exceptions. MLX also has a
fully featured C++ API which closely follows the Python API. fully featured C++ API which closely follows the Python API.
@ -17,7 +17,7 @@ The main differences between MLX and NumPy are:
- **Multi-device**: Operations can run on any of the supported devices (CPU, - **Multi-device**: Operations can run on any of the supported devices (CPU,
GPU, ...) GPU, ...)
The design of MLX is strongly inspired by frameworks like `PyTorch The design of MLX is inspired by frameworks like `PyTorch
<https://pytorch.org/>`_, `Jax <https://github.com/google/jax>`_, and <https://pytorch.org/>`_, `Jax <https://github.com/google/jax>`_, and
`ArrayFire <https://arrayfire.org/>`_. A noteable difference from these `ArrayFire <https://arrayfire.org/>`_. A noteable difference from these
frameworks and MLX is the *unified memory model*. Arrays in MLX live in shared frameworks and MLX is the *unified memory model*. Arrays in MLX live in shared