ml-explore/mlx-swift-examples

해결하는 문제

이 저장소는 MLX Swift를 사용하는 개발자가 Apple silicon에서 AI 모델과 수치 계산 작업을 구현하는 방법을 확인할 수 있도록 실용적인 예제와 재사용 가능한 라이브러리의 컬렉션을 제공합니다. 원시적인 MLX Swift 프레임워크와 iOS 및 macOS에서 실제 애플리케이션을 개발하는 사이의 격차를 메웁니다.

작동 방식

프로젝트는 AI 애플리케이션(채팅 앱 및 이미지 생성기 등), 빠른 테스트를 위한 명령줄 도구, 수치 계산 시뮬레이션의 세 가지 주요 범주로 예제를 정리합니다. 언어 모델에는 mlx-swift-lm 라이브러리를 활용하고, Stable Diffusion 및 MNIST에 대해서는 자체 구현을 제공합니다.

대상 사용자

Apple 플랫폼(iOS 및 macOS)용 AI 기반 애플리케이션을 MLX 프레임워크를 사용해 개발하고자 하는 Swift 개발자.

주요 특징

• 다양한 AI 예제: LLM, VLM, 이미지 생성을 위한 Stable Diffusion의 구현 포함.
• 크로스플랫폼: 많은 예제가 iOS와 macOS 모두에서 실행 가능.
• 피니튜닝: macOS에서 LLM을 피니튜닝하기 위한 LoRA 학습 예제 포함.
• 수치 계산: 커스텀 Metal 커널을 사용하여 열 확산 및 맨델브로 집합 렌더링과 같은 일반적인 수치 작업을 시연.
• 즉시 사용 가능한 도구: Hugging Face의 모델을 빠르게 테스트할 수 있는 텍스트 및 이미지 생성용 CLI 도구 제공.