From b9e7113a6b9f6203e700db1be768e296dda7d8ea Mon Sep 17 00:00:00 2001 From: Fizz~ <168598314+fizzAI@users.noreply.github.com> Date: Fri, 23 Jan 2026 19:26:53 -0500 Subject: [PATCH] Update README with CUDA support, limitations, screenshots --- README.md | 16 ++++++++++++++-- 1 file changed, 14 insertions(+), 2 deletions(-) diff --git a/README.md b/README.md index 30116fa..488c683 100644 --- a/README.md +++ b/README.md @@ -2,7 +2,12 @@ **Status: WIP / Experimental** -The goal of this project is to provide MLX training on Apple Silicon with semi-drop-in support for Axolotl configurations (best-effort). +The goal of this project is to provide MLX training on Apple Silicon (and CUDA devices) with semi-drop-in support for Axolotl configurations (best-effort). + +

+image +image +

## Setup @@ -30,4 +35,11 @@ uv run black . # Lint code uv run ruff check . -``` \ No newline at end of file +``` + +## Limitations on CUDA + +Due to MLX limitations, some features are not supported on CUDA devices: + +- QLoRA (4bit and 8bit) +- Possibly other stuff!