Support dot broadcast - Githubissues

SciML / ADTypes.jl

Repository for automatic differentiation backend types

https://sciml.github.io/ADTypes.jl/

MIT License

36 stars 11 forks source link

Support dot broadcast #39

Closed prbzrg closed 3 months ago

prbzrg commented 3 months ago

Checklist

[x] Appropriate tests were added
[x] Any code changes were done in a way that does not break public API
[x] All documentation related to code changes were updated
[x] The new code follows the contributor guidelines, in particular the SciML Style Guide and COLPRAC.
[x] Any new documentation only uses public API

Additional context

We need this patch to be able to use something like:

DifferentiationInterface.jacobian.(x -> Lux.apply(nns, x, ps), AutoZygote(), eachcol(r))