gaurav-arya / StochasticAD.jl

Research package for automatic differentiation of programs containing discrete randomness.
MIT License
198 stars 16 forks source link

Support derivatives of functions with vector inputs #40

Closed slwu89 closed 1 year ago

slwu89 commented 1 year ago

Very small PR to address #32; @AlCap23 I didn't put in any of your comments on that issue so you can do a PR and take proper credit. Plus your Julia looks much nicer than mine!

gaurav-arya commented 1 year ago

@slwu89 can you let me know if the new interface works for you?

slwu89 commented 1 year ago

@slwu89 can you let me know if the new interface works for you?

Yes! Looks great.