Closed GilesStrong closed 1 year ago
Swaps out pytorch internal vmap for functorch vmap: same functionality, but more performant implementation. Adds functorch as dependency.
Swaps out pytorch internal vmap for functorch vmap: same functionality, but more performant implementation. Adds functorch as dependency.