Allow for more elemwise torch functions using broadcast_tensor and vmap#1032
Merged
ricardoV94 merged 1 commit intopymc-devs:mainfrom Ch0ronomato:elemwise_torch_improvementNov 19, 2024
+48-3
broadcast_tensor and vmap#1032