MLX (Apple ML framework)
| MLX | |
|---|---|
| Developer | Apple |
| Release | December 5, 2023 |
| Written in | C++, Python, CUDA, Metal |
| Type | Machine learning framework |
| License | MIT License |
| Website | mlx-framework |
| Repository | https://github.com/ml-explore/mlx |
MLX is an open-source machine learning framework developed by Apple and designed primarily for Apple silicon. It is used for training and running machine learning models and is optimized for Apple silicon's unified memory architecture, allowing the CPU and GPU to work with the same data without copying it between separate memory pools. MLX can be used for applications including large language model training and inference, image generation, and speech recognition.