A framework for machine learning on Apple silicon.

Navigation

Verified details

These details have been verified by PyPI
Project links
Owner
GitHub Statistics
Maintainers
Avatar for awni from gravatar.com awni Avatar for katharas from gravatar.com katharas Avatar for mlx-dev from gravatar.com mlx-dev

Unverified details

These details have not been verified by PyPI
Meta
Report project as malware

Project description

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:

  • Familiar APIs: MLX has a Python API that closely follows NumPy. MLX also has fully featured C++, C, and Swift APIs, which closely mirror the Python API. MLX has higher-level packages like mlx.nn and mlx.optimizers with APIs that closely follow PyTorch to simplify building more complex models.

  • Composable function transformations: MLX supports composable function transformations for automatic differentiation, automatic vectorization, and computation graph optimization.

  • Lazy computation: Computations in MLX are lazy. Arrays are only materialized when needed.

  • Dynamic graph construction: Computation graphs in MLX are constructed dynamically. Changing the shapes of function arguments does not trigger slow compilations, and debugging is simple and intuitive.

  • Multi-device: Operations can run on any of the supported devices (currently the CPU and the GPU).

  • Unified memory: A notable difference from MLX and other frameworks is the unified memory model. Arrays in MLX live in shared memory. Operations on MLX arrays can be performed on any of the supported device types without transferring data.

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},
}

Project details

Verified details

These details have been verified by PyPI
Project links
Owner
GitHub Statistics
Maintainers
Avatar for awni from gravatar.com awni Avatar for katharas from gravatar.com katharas Avatar for mlx-dev from gravatar.com mlx-dev

Unverified details

These details have not been verified by PyPI
Meta

Release history Release notifications | RSS feed

This version

0.31.2

0.31.1

0.31.0

0.30.6

0.30.5

0.30.4

0.30.3

0.30.1

0.30.0

0.29.4

0.29.3

0.29.2

0.29.1

0.29.0

0.28.0

0.27.1

0.26.5

Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distributions

No source distribution files available for this release.See tutorial on generating distribution archives.

Built Distributions

If you're not sure about the file name format, learn more about wheel file names.

mlx_metal-0.31.2-py3-none-macosx_26_0_arm64.whl (55.8 MB view details)

Uploaded Python 3macOS 26.0+ ARM64

mlx_metal-0.31.2-py3-none-macosx_15_0_arm64.whl (40.1 MB view details)

Uploaded Python 3macOS 15.0+ ARM64

mlx_metal-0.31.2-py3-none-macosx_14_0_arm64.whl (40.1 MB view details)

Uploaded Python 3macOS 14.0+ ARM64

File details

Details for the file mlx_metal-0.31.2-py3-none-macosx_26_0_arm64.whl.

File metadata

File hashes

Hashes for mlx_metal-0.31.2-py3-none-macosx_26_0_arm64.whl
Algorithm Hash digest
SHA256 84ffb60ee503f03eb684f5fb168d5cff31e2a16b7f27c1731eaf7662bd6e9b46
MD5 14a17633cd5c0625caa99370ba8f8395
BLAKE2b-256 998211fd62a8d7a3e96e5c43220b17de0151e3f10101f8bb3b865f5bd9cdd074

See more details on using hashes here.

Provenance

The following attestation bundles were made for mlx_metal-0.31.2-py3-none-macosx_26_0_arm64.whl:

Publisher: release.yml on ml-explore/mlx

Attestations: Values shown here reflect the state when the release was signed and may no longer be current.

File details

Details for the file mlx_metal-0.31.2-py3-none-macosx_15_0_arm64.whl.

File metadata

File hashes

Hashes for mlx_metal-0.31.2-py3-none-macosx_15_0_arm64.whl
Algorithm Hash digest
SHA256 e9d4e5fce6ca10a87a0e388597f99519ad594d09e674708b5312bd8bd4f5997d
MD5 00fe2fba39568c95cd9db52b7bc63465
BLAKE2b-256 4f5d4c690d5b93c30ba002656c37363159d978705bf8eb801b8481840fb942c2

See more details on using hashes here.

Provenance

The following attestation bundles were made for mlx_metal-0.31.2-py3-none-macosx_15_0_arm64.whl:

Publisher: release.yml on ml-explore/mlx

Attestations: Values shown here reflect the state when the release was signed and may no longer be current.

File details

Details for the file mlx_metal-0.31.2-py3-none-macosx_14_0_arm64.whl.

File metadata

File hashes

Hashes for mlx_metal-0.31.2-py3-none-macosx_14_0_arm64.whl
Algorithm Hash digest
SHA256 b25385bcee18fc194092255b8b53b9a3d8489eb650e59160f1b57aadd07aa2dc
MD5 ae2bacad0e3efbbab39a8c199406f5b6
BLAKE2b-256 3f69fe3b783ebe999f3118234e1e940feb622518bfb1dea6ac5d13b1d36a8449

See more details on using hashes here.

Provenance

The following attestation bundles were made for mlx_metal-0.31.2-py3-none-macosx_14_0_arm64.whl:

Publisher: release.yml on ml-explore/mlx

Attestations: Values shown here reflect the state when the release was signed and may no longer be current.

Supported by

AWS Cloud computing and Security Sponsor Datadog Monitoring Depot Continuous Integration Fastly CDN Google Download Analytics Pingdom Monitoring Sentry Error logging StatusPage Status page