Joe Cummings

Fix import of `torchao` now that `torchao-nightly` does not exist

1

[Remove torchao_nightly package by atalman · Pull Request #661 · pytorch/ao](https://l.workplace.com/l.php?u=https%3A%2F%2Fgithub.com%2Fpytorch%2Fao%2Fpull%2F661&h=AT1jwbc7yG3vKKLkcD_rFWX0NRhMn6ve5q4k06-hNUtYT1naRDq00lmq0q0WJr2tqsY9KtHhUYGs4BWRTw6W43AU5i_XFbF__-bUh0ScmNsQ8a4k8M-c3i0054jKe9bTHwDpKb6d8Unet39lrKLBQcDKZXFJyEAQcw)

MPS Support

6

Copy of #790 but with proper rebase @maximegmd is the real hero here.

Add more coding models

1

* Codestral * CodeGemma * CodeQwen * Deepseek

In the docs, we mention that the model_transform is expected to return "tokens" and "mask". We should add better validation that these are the returned values and blame the transform....

Very unclear how to get to Message Transforms from SFTDataset

https://github.com/tensorflow/tensorboard/issues/6869#issuecomment-2289728902

Most people use safetensors, we shouldn't expect people to understand that we ignore them by default. This will also require us to update our default download commands across the docs...

Pls.