accelerate
0e4419b3 - Add bf16/fp16 support for amp with mps device (#3373)

Commit
133 days ago
Add bf16/fp16 support for amp with mps device (#3373) * Fix tests * format * amp mps support for fp16/bf16 * add error * revert * revert * fix * ruff
Author
Parents
Loading