search

graphcore-research / unit-scaling

A library for unit scaling in PyTorch
https://graphcore-research.github.io/unit-scaling/
Apache License 2.0
104 stars 7 forks source link

Update to PyTorch 2.2 addendum: Fixing a doc typo and logic mismatch #54

Closed awf closed 6 months ago

awf commented 6 months ago

I realised there was a doc typo, which led me to also make a logic change to better match the new doc link.