Note about MPS in ch06 and ch07

rasbt / LLMs-from-scratch

Implementing a ChatGPT-like LLM in PyTorch from scratch, step by step

Other

27.86k stars 3.16k forks source link

Closed rasbt closed 1 month ago

rasbt commented 1 month ago

Adds notes about MPS to ch06 and ch07 since it seems to be much more usable now in PyTorch 2.4

review-notebook-app[bot] commented 1 month ago

Check out this pull request on

See visual diffs & provide feedback on Jupyter Notebooks.