MLX — Apple 실리콘 머신러닝 프레임워크

MLX는 Apple 실리콘(M 시리즈) 을 위해 Apple 머신러닝 리서치팀이 만든 배열(array) 프레임워크예요. NumPy처럼 친숙한 API를 제공하면서도, Apple 칩의 통합 메모리를 그대로 활용해서 학습과 배포가 모두 빠릅니다.

출처: https://github.com/ml-explore/mlx

MLX의 가장 큰 특징은 Unified Memory(통합 메모리) 모델이에요. 배열이 CPU와 GPU가 공유하는 메모리에 살기 때문에 데이터를 옮길 필요 없이 어느 디바이스에서든 연산할 수 있죠. 그래서 노트북 한 대에서도 LLM을 직접 돌리거나 파인튜닝하는 실험이 흔해졌어요.