mirror of
https://github.com/priyanshujain/torchmlx.git
synced 2026-10-02 11:07:13 +00:00
update readme
This commit is contained in:
1 parent
53c9cf4159
commit
b5f11ad937
1 file changed
+4
-4
@@ -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
|
||||
import torchmlx as torch
|
||||
@@ -12,4 +12,4 @@ 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).
|
||||
see the [tinystories example](examples/tinystories-llm/train.py) and [compatibility details](docs/compatibility.md).
|
||||
Reference in new issue
Block a user