pj/torchmlx
1
0 Fork 0
mirror of https://github.com/priyanshujain/torchmlx.git synced 2026-10-02 11:07:13 +00:00
Code Issues Packages Projects Releases Wiki Activity

update readme

Browse Source
This commit is contained in:
pj committed 2026-09-19 23:17:48 +05:30
1 parent 53c9cf4159
commit b5f11ad937
1 file changed
+4 -4
Download Patch File Download Diff File Expand all files Collapse all files
README.md
+4 -4
View File
@@ -1,8 +1,8 @@
# TorchMLX # torchmlx
TorchMLX is a PyTorch-shaped compatibility layer that uses MLX on Apple silicon and PyTorch elsewhere. torchmlx is a pytorch-shaped compatibility layer that uses mlx on apple silicon and pytorch elsewhere.
It is experimental and targets transformer inference and training without per-operation backend fallback. NOTE: it is experimental and built first for educational use. If you find a bug, please create an issue on the repo.
```python ```python
import torchmlx as torch import torchmlx as torch
@@ -12,4 +12,4 @@ model = nn.Linear(4, 2)
optimizer = optim.AdamW(model.parameters(), lr=3e-4) optimizer = optim.AdamW(model.parameters(), lr=3e-4)
``` ```
See the [TinyStories example](examples/tinystories-llm/train.py) and [compatibility details](docs/compatibility.md). see the [tinystories example](examples/tinystories-llm/train.py) and [compatibility details](docs/compatibility.md).