search

pytorch / functorch

functorch is JAX-like composable function transforms for PyTorch.
https://pytorch.org/functorch/
BSD 3-Clause "New" or "Revised" License
1.38k stars 102 forks source link

fix logsigmoid_backward batching rule #1033

Closed zou3519 closed 1 year ago

zou3519 commented 1 year ago

https://github.com/pytorch/pytorch/pull/84892 will fix this

zou3519 commented 1 year ago

Completed