mirror of
https://github.com/ml-explore/mlx-examples.git
synced 2025-12-16 02:08:55 +08:00
Update README.md
This commit is contained in:
@@ -111,7 +111,7 @@ Benchmark PyTorch MPS performance:
|
|||||||
python benchmarks/benchmark_pt.py
|
python benchmarks/benchmark_pt.py
|
||||||
```
|
```
|
||||||
|
|
||||||
Expected performance on M4 MacBook Pro (batch_size = 5):
|
Expected performance on M4 MacBook Pro (ESM-2 650M, batch_size = 5):
|
||||||
|
|
||||||
- MLX: 299 ms per step, 16.71 sequences/sec
|
- MLX: 299 ms per step, 16.71 sequences/sec
|
||||||
- PyTorch MPS: 402 ms per step, 12.43 sequences/sec
|
- PyTorch MPS: 402 ms per step, 12.43 sequences/sec
|
||||||
|
|||||||
Reference in New Issue
Block a user