search

anucvml / ddn

Deep Declarative Networks
MIT License
237 stars 37 forks source link

Future Feature Request: Vectorization #7

Open Parskatt opened 3 years ago

Parskatt commented 3 years ago

Pytorch is (slowly) introducing vmap: https://pytorch.org/docs/master/generated/torch.vmap.html?highlight=vmap#torch.vmap When this feature becomes stable it seems like a great addition for jacobian calculation, probably giving additional performance.

sgould commented 3 years ago

This is cool. Thanks for recommending. Once vmap makes it into an official release we'll look to profile and use it.