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!