mlx README

The project's README from github.com/ml-explore/mlx. As one markdown file: README.md.

MLX

Quickstart | Installation | Documentation | Examples

CircleCI

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:

MLX is designed by machine learning researchers for machine learning researchers. The framework is intended to be user-friendly, but still efficient to train and deploy models. The design of the framework itself is also conceptually simple. We intend to make it easy for researchers to extend and improve MLX with the goal of quickly exploring new ideas.

The design of MLX is inspired by frameworks like NumPy, PyTorch, Jax, and ArrayFire.

Examples

The MLX examples repo has a variety of examples, including:

Quickstart

See the quick start guide in the documentation.

Installation

MLX is available on PyPI. To install MLX on macOS, run:

pip install mlx

To install the CUDA backend on Linux, run:

pip install mlx[cuda]

To install a CPU-only Linux package, run:

pip install mlx[cpu]

Checkout the documentation for more information on building the C++ and Python APIs from source.

Contributing

Check out the contribution guidelines for more information on contributing to MLX. See the docs for more information on building from source, and running tests.

We are grateful for all of our contributors. If you contribute to MLX and wish to be acknowledged, please add your name to the list in your pull request.

Citing MLX

The MLX software suite was initially developed with equal contribution by Awni Hannun, Jagrit Digani, Angelos Katharopoulos, and Ronan Collobert. If you find MLX useful in your research and wish to cite it, please use the following BibTex entry:

@software{mlx2023,
  author = {Awni Hannun and Jagrit Digani and Angelos Katharopoulos and Ronan Collobert},
  title = {{MLX}: Efficient and flexible machine learning on Apple silicon},
  url = {https://github.com/ml-explore},
  version = {0.0},
  year = {2023},
}

mlx (ml-explore/mlx)

Searching Database…

28.8K ·GitHub
Что это такое?MLX — это фреймворк для работы с массивами в машинном обучении на чипах Apple silicon, созданный командой исследований машинного обучения Apple.
Это Python-библиотека или полноценный фреймворк?И то и другое: есть Python API, очень похожий на NumPy, плюс API для C++, C и Swift, повторяющие его, а также пакеты более высокого уровня mlx.nn и mlx.optimizers, сделанные по образцу PyTorch.
Заработает ли это на моём компьютере или только на Mac?В первую очередь это Mac на чипах Apple silicon, но Linux тоже поддерживается — через CUDA-бэкенд или пакет только для CPU. Поддержка Windows нигде не упоминается.
Чем MLX отличается от NumPy или PyTorch?Своей моделью единой памяти: массивы лежат в общей памяти, поэтому операции выполняются на CPU или GPU без копирования данных. Вычисления ленивые, графы строятся динамически, а автоматическое дифференцирование и векторизация встроены с самого начала.
Что на нём реально можно сделать?В репозитории с примерами есть обучение языковых моделей на трансформерах, генерация текста в больших масштабах с LLaMA и дообучение с LoRA, генерация изображений со Stable Diffusion и распознавание речи с Whisper.
Почему Apple создала это, вместо того чтобы просто использовать PyTorch?Его делали исследователи машинного обучения для исследователей: инструмент дружелюбный к пользователю, но при этом эффективный для обучения и развёртывания моделей и концептуально простой для расширения, чтобы быстро проверять новые идеи. На дизайн повлияли NumPy, PyTorch, Jax и ArrayFire.
Creator
ml-explore
Size
69.4 MB
Stars
28.8K
Age
2y
Commits · 90 Days