Skip to content

Allow for more elemwise torch functions using broadcast_tensor and vmap#1032

Merged
ricardoV94 merged 1 commit intopymc-devs:mainfrom
Ch0ronomato:elemwise_torch_improvement
Nov 19, 2024
Merged

Allow for more elemwise torch functions using `broadcast_tensor` and `vmap`#1032
ricardoV94 merged 1 commit intopymc-devs:mainfrom
Ch0ronomato:elemwise_torch_improvement

Commits

Commits on Nov 6, 2024