Autograd + expamles revisiting

ITMO-NSS-team / torch_DE_solver

Combines power of torch, numerical methods to conquer and solve ALL {O,P}DEs

https://torch-de-solver.readthedocs.io

BSD 3-Clause "New" or "Revised" License

59 stars 9 forks source link

Autograd + expamles revisiting #6

Closed SuperSashka closed 2 years ago

SuperSashka commented 2 years ago

I add the torch.autograd functional, such that the pytorch models are differentiated with AD. Also all examples are updated and should be working using any IDE.