swift-mlx

What problem does it solve?

This Skill provides guidance and examples for using the MLX Swift framework to build and train machine learning models efficiently on Apple Silicon hardware.

Core Features & Use Cases

• Array Operations: Perform high-performance numerical computations with MLXArrays.
• Neural Network Building: Define and utilize layers from MLXNN for various architectures.
• Model Training: Implement training loops with automatic differentiation and optimizers.
• Use Case: Develop and fine-tune a custom neural network for image classification directly on your Mac using Swift.

Quick Start

Use the swift-mlx skill to create a basic MLXArray and perform element-wise addition.