ml-explore/mlx-swift

해결하는 문제

MLX Swift는 MLX 배열 프레임워크용 네이티브 Swift API를 제공하여 Apple silicon에서 Swift 언어 내에서 직접 기계학습 연구 및 실험을 수행할 수 있도록 합니다.

작동 방식

MLX 프레임워크의 Swift 확장으로 작동하며, 사용자가 Swift를 사용하여 모델을 구축하고 훈련할 수 있습니다. Xcode, SwiftPM, CMake를 포함한 여러 빌드 시스템을 지원하며, macOS/iOS에서는 Metal, Linux에서는 CUDA 백엔드를 제공합니다.

대상 사용자

macOS, iOS, Linux에서 Apple silicon의 성능을 활용해 기계학습 작업을 수행하고자 하는 Swift 개발자 및 기계학습 연구자.

주요 특징

• MLX 배열 프레임워크용 네이티브 Swift API.
• 보완 패키지 mlx-swift-lm을 통해 LLM 및 VLM 구현을 지원.
• macOS, iOS, Linux에서의 크로스플랫폼 지원.
• MNIST에 대해 LeNet을 훈련시키는 예제, LLM/VLM을 사용한 채팅 앱 실행, Stable Diffusion을 통한 이미지 생성 예제 포함.