Closed IgnacioJPickering closed 2 years ago
@zasdfgbnm calling without as_tuple is deprecated... but as_tuple is not jittable?
Errr, that's bad. Maybe you can file an issue to PyTorch?
pytorch/pytorch#40601 I'll see what they say
This appears to be the relevant PyTorch issue now: https://github.com/pytorch/pytorch/issues/45499
The way we are calling nonzero in AEV computations is deprecated in the latest pytorch build