apple Project

MLX

About

MLX is an array framework designed for efficient and flexible machine learning research on Apple silicon.

MLX is an array framework optimized for the unified memory architecture of Apple silicon. The NumPy-like API makes it familiar to use and flexible. The higher level neural net and optimizer packages along with function transformations for automatic differentiation and graph optimization let you build more complex yet efficient machine learning models. MLX also has Swift, C++, and C bindings and can run on any Apple platform.

Language

  • Python
  • Swift
  • C
  • C++

Links