build torchmlx

This commit is contained in:
pj committed 2026-09-19 22:52:14 +05:30
commit b0ef50eec2
15 files changed
+2008

No files matched your search

+15
View File
@@ -0,0 +1,15 @@
# TorchMLX
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.
```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).