Skip to content

lhuang-pvamu/pytorch-examples

Repository files navigation

pytorch-examples

  1. simple_ad.py: a simple auto differentiation code using Pytorch

  2. simple_lr.py: a simple linear regression code using Pytorch module

  3. fluidsim.py: a fluid simulation using auto differentiation to optimize its velocity model.

  4. sin.py: sin(x) and its gradient verification as cos(x). An example from the Julia paper: A differentiable programming system...

About

Pytorch examples, especially for auto differentiation

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages