clean up torchmlx internals

This commit is contained in:
pj committed 2026-09-20 16:49:44 +05:30
1 parent 4eb86111a8
commit cd140fae7a
19 files changed
+1115 -794

No files matched your search

+5
View File
@@ -12,3 +12,8 @@ from torchmlx import nn, optim
```
see the [tinystories example](examples/tinystories-llm/train.py) and [compatibility details](docs/compatibility.md).
| backend | 5,000 steps + generation | effective steps/s |
| --- | ---: | ---: |
| torchmlx mlx, m3 pro | 170.27 s | 29.4 |
| pytorch mps, m3 pro | 184.42 s | 27.1 |