Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Support autodiff for ndarray #4267

Open
erizmr opened this issue Feb 14, 2022 · 1 comment
Open

Support autodiff for ndarray #4267

erizmr opened this issue Feb 14, 2022 · 1 comment
Assignees
Labels
autodiff This issue is related to automatic differentiation system feature request Suggest an idea on this project

Comments

@erizmr
Copy link
Contributor

erizmr commented Feb 14, 2022

Concisely describe the proposed feature
Currently, the autodiff system only supports Taichi field. However, there are some limitations of field such as the number of snode trees is limited which blocks the scalability of the applications. While Ndarray is a container that has more flexibility and is currently supported by multiple backends of Taichi. Support autodiff for Ndarray will broad the application of the autodiff system.

Describe the solution you'd like (if any)

  1. Add grad for ndarray (include an adjoint array etc. )
  2. Include ExternalPtrStmt in source code transformation of autodiff.
@erizmr erizmr added feature request Suggest an idea on this project autodiff This issue is related to automatic differentiation system labels Feb 14, 2022
@erizmr erizmr self-assigned this Feb 14, 2022
@chenzhekl
Copy link

It seems that the taichi-nerfs codebase is already using some sort of autodiff for ndarray. Does that imply that this functionality is already working in the nightly version?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
autodiff This issue is related to automatic differentiation system feature request Suggest an idea on this project
Projects
None yet
Development

No branches or pull requests

2 participants