perf(transforms): Use stop_gradient in custom_function VJP#2951
Closed
Rudra-Tiwari-codes wants to merge 1 commit intoml-explore:mainfrom
Rudra-Tiwari-codes:perf/optimize-custom-vjp
Closed
perf(transforms): Use stop_gradient in custom_function VJP#2951Rudra-Tiwari-codes wants to merge 1 commit intoml-explore:mainfrom Rudra-Tiwari-codes:perf/optimize-custom-vjp
Rudra-Tiwari-codes wants to merge 1 commit intoml-explore:mainfrom
Rudra-Tiwari-codes:perf/optimize-custom-vjp
