A lightweight machine learning framework based on MLX with an API inspired by FastEstimator.
This repository contains a minimal implementation sufficient to run the MNIST example in apphub/mnist.py
and a simple CIFAR-10 variant in apphub/cifar10.py
.
Install fastmlx
using pip
:
pip install -e .
The unit tests use Python's unittest
framework. Run all tests with:
python -m unittest discover -s tests