Swift team announces ML research project

Swift team announces MLX Swift – framework that brings MLX capabilities to Swift.

MLX has several important features for machine learning research that few if any existing Swift libraries support. These include:

• Native support for hardware acceleration. MLX can run compute intensive operations on the CPU or GPU.
• Automatic differentiation for training neural networks and the gradient-based machine learning models

The documentation contains a few more complete examples to help you get started with MLX Swift:

• Text generation with an LLM: A complete LLM text generation example with Mistral 7B. The example will generate text using any Mistral or Llama-style model including pre-quantized MLX models, many of which are available on Hugging Face.
• Training an MLP on MNIST: The example trains a simple multi-layer perceptron to classify MNIST digits using the MLX Swift neural network and optimizers packages.

Swift Blog

GitHub: mlx-swift

Swift Blog: On-device ML research with MLX and Swift

Reference: Apple Silicon gets massive AI training speed boost with this new project