Tags: grievejia/pytorch
Tags
Revert "Add nested tensor support to autograd (pytorch#79446)" This reverts commit f965681.
Update on "Add forward AD for linalg.det and simplify its backward" This PR is in preparation for implementing `logdet` and `slogdet` as structured kernels + implementing them with more efficient derivatives We implement forward AD for det. We also simplify the implementation of the backward, and leave a note on how to implement it properly for singular matrices. We leave thad for future work. Note (by looking at the OpInfo) that the current implementation passes the same tests as the one before. We skip the forward-over-backward in the singular case, as that one was not working in the gradgrad case either. [ghstack-poisoned]
[2] move more pytorch buck targets to shared build (pytorch#79330) Summary: Pull Request resolved: pytorch#79330 Move more xplat buck targets to the shared build file Test Plan: sandcastle and oss CI Differential Revision: D37087229 fbshipit-source-id: c0efed4a86e253b0732e682c7c6955c9d7e6213c
Update on "Add forward AD for linalg.det and simplify its backward" This PR is in preparation for implementing `logdet` and `slogdet` as structured kernels + implementing them with more efficient derivatives We implement forward AD for det. We also simplify the implementation of the backward, and leave a note on how to implement it properly for singular matrices. We leave thad for future work. Note (by looking at the OpInfo) that the current implementation passes the same tests as the one before. We skip the forward-over-backward in the singular case, as that one was not working in the gradgrad case either. [ghstack-poisoned]
reland Add offsets-based reduction to segment_reduce (CPU, CUDA) [ghstack-poisoned]
Add amax and amin with tests (pytorch#33) * Add amax and amin with tests
PreviousNext