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

publish pytorchmlx

Browse Source
This commit is contained in:
pj committed 2026-09-19 23:47:23 +05:30
1 parent cf90000071
commit fefdea2304
5 files changed
+35 -33
Download Patch File Download Diff File Expand all files Collapse all files

No files matched your search

README.md
+4 -5
View File
@@ -1,15 +1,14 @@
# 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 built first for educational use.
NOTE: it is experimental and built first for educational use. If you find a bug, please create an issue on the repo.
```bash
pip install pytorchmlx
```
```python
import torchmlx as torch
from torchmlx import nn, optim
model = nn.Linear(4, 2)
optimizer = optim.AdamW(model.parameters(), lr=3e-4)
```
see the [tinystories example](examples/tinystories-llm/train.py) and [compatibility details](docs/compatibility.md).