mirror of
https://github.com/priyanshujain/torchmlx.git
synced 2026-10-02 11:07:13 +00:00
clean up torchmlx internals
This commit is contained in:
1 parent
4eb86111a8
commit
cd140fae7a
19 files changed
+1115
-794
No files matched your search
@@ -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 |
|
||||
Reference in new issue
Block a user