This project is an implementation of a convolutional neural network (CNN) using the MLX framework to classify handwritten digits in the MNIST dataset.
python3 -m venv venv
source venv/bin/activate
pip install -r requirements
python app.py
# run the pytorch example
python pytorch_example/app.py
We use MLflow to track accuracies and losses during model training and evaluation
Thanks to mikecvet for this implementation, which greatly assisted in the development of this project.