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

Conv1d #67

Closed thecharlieblake closed 2 months ago

thecharlieblake commented 2 months ago

Implementing a Conv1d function and module, building on #66 by @norikazu99.