anucvml / ddn

Deep Declarative Networks
MIT License
237 stars 37 forks source link

Update linear algebra routines #11

Closed sgould closed 1 year ago

sgould commented 3 years ago

Update code to use linear algebra routines in pytorch.linalg and support pytorch v1.9.0 and above only.