Closed gaurav-arya closed 5 months ago
Addresses #1387
These new rules could potentially be too narrow for some cases. Also, for reference, here are the ones in ChainRules.jl: https://github.com/JuliaDiff/ChainRules.jl/blob/f9f0722558e065ee8bbdf042c2d99a9d56ba0070/src/rulesets/Random/random.jl#L15-L25. It seemed like we might not need some of them, and also the generic rand! ones there would be dangerous for Enzyme when differentiating parameter-dependent mutable sampling.
ChainRules.jl
rand!
Addresses #1387
These new rules could potentially be too narrow for some cases. Also, for reference, here are the ones in
ChainRules.jl: https://github.com/JuliaDiff/ChainRules.jl/blob/f9f0722558e065ee8bbdf042c2d99a9d56ba0070/src/rulesets/Random/random.jl#L15-L25. It seemed like we might not need some of them, and also the genericrand!ones there would be dangerous for Enzyme when differentiating parameter-dependent mutable sampling.