functorch
functorch copied to clipboard
pytorch
functorch is JAX-like composable function transforms for PyTorch.
Hi, I am trying to compile functorch from source on a Jetson platform. Specifically, it is Jetpack 5.0.2 on NVIDIA Xavier AGX. I install PyTorch by following the instructions at...
## Problem Today, some of our batching rules need to duplicate error checking code. For example, consider https://github.com/pytorch/pytorch/pull/89384, which adds a check to the structured kernel that the input to...
Hi, thanks for the great library! I met the following issue when trying to install functorch: 1. in the doc, it says functorch is already included in pytorch 1.13, but...
Hi, I implemented a Jacobian computation using functorch, but encoutnered a memory overflow issue. The function that I want to differentiate is `ResidualFunctional.residual`. I'd like to compute the Jacobian of...
Hi, I'd like to use `functorh` to realize following loss: ### Question demonstrate assume the - the dimension of output tensor is $O$ and we will use $y^\gamma$ mark each...
Hi, experts. I am currently calculating a Jacobian column-by-column and calculating the squared sum of each column to calculate the Trace of the Jacobian. The code looks something like this:...
Hi All, First, just a quick note, you will need pennylane installed in order to run this toy example. Here is the code that I am trying to execute: ```...
I have a function ```tridiagonalization``` to tridiagonalize matrix (2d tensor), and I want to map it to batch. It involves a for loop and on each iteration a permutation of...
Hello, I am using [this linked solution](https://stackoverflow.com/questions/78076239/how-to-efficiently-calculate-gradients-of-all-outputs-with-respect-to-parameters/78087089#78087089) from stack overflow to compute gradients more efficiently than a manual loop. I notice that there is some small difference in the gradients...
Hello everyone, I need to implement a VMAP over a complex function that at some point calls a standard [torch.GRU](https://pytorch.org/docs/stable/generated/torch.nn.GRU.html). However, when the VMAP function is called, a "RuntimeError: Batching...
