Search results

VMAP over GRU: Batching rule not implemented for aten::gru.input

Status: Open.

#1147 In pytorch/functorch;

· BernardoFerreira opened on Aug 27, 2024

Tensor.nonzero_static fails on GPU inside torch.func.vmap

Status: Open.

#1145 In pytorch/functorch;

· Tendocat opened on Jul 15, 2024

Small difference between functorch grads and torch.autograd.grad

Status: Open.

#1144 In pytorch/functorch;

· alanjeffares opened on May 1, 2024

vmap does not support Tensor.clone()

Status: Open.

#1143 In pytorch/functorch;

· JustinS6626 opened on Apr 10, 2024

Swapping 2 columns in a 2d tensor

Status: Open.

#1142 In pytorch/functorch;

· Kreativshikkk opened on Mar 13, 2024

wrapper->level().value() <= current_level INTERNAL ASSERT FAILED at "../aten/src/ATen/functorch/ADInterpreters.cpp":39

Status: Open.

#1131 In pytorch/functorch;

· ElegantLin opened on Jun 7, 2023

Per-sample-gradient: Get gradient 0 when using grad(params_tograd, params) with respect to part of model's parameters

Status: Open.

#1122 In pytorch/functorch;

· Ancientshi opened on Apr 1, 2023

How to get the jacobian matrix in GCNs?

Status: Open.

#1113 In pytorch/functorch;

· pcheng2 opened on Feb 27, 2023

Error about using a grad transform with in-place operation is inconsistent with and without DDP

Status: Open.

#1112 In pytorch/functorch;

· XuchanBao opened on Feb 24, 2023

Use functional models inside usual nn.Module

Status: Open.

#1111 In pytorch/functorch;

· subho406 opened on Feb 15, 2023

Performance drop because of not yet implemented batching rule for bincount

Status: Open.

#1109 In pytorch/functorch;

· IosiaLectus opened on Feb 9, 2023

item() support for vmap

Status: Open.

#1108 In pytorch/functorch;

· baskuit opened on Jan 28, 2023