Add autodiff feature
Showing
- pystencils/autodiff/__init__.py 14 additions, 0 deletionspystencils/autodiff/__init__.py
- pystencils/autodiff/_autodiff_astpair.py 34 additions, 0 deletionspystencils/autodiff/_autodiff_astpair.py
- pystencils/autodiff/adjoint_field.py 31 additions, 0 deletionspystencils/autodiff/adjoint_field.py
- pystencils/autodiff/autodiff.py 539 additions, 0 deletionspystencils/autodiff/autodiff.py
- pystencils/autodiff/backends.py 25 additions, 0 deletionspystencils/autodiff/backends.py
- pystencils_tests/autodiff/test_autodiff.py 51 additions, 0 deletionspystencils_tests/autodiff/test_autodiff.py
- pystencils_tests/autodiff/test_tfmad.py 246 additions, 0 deletionspystencils_tests/autodiff/test_tfmad.py
Please register or sign in to comment