EnzymeAD / rust

A rust fork to work towards Enzyme integration
https://www.rust-lang.org
Other
53 stars 7 forks source link

Add vectorization support #53

Open ZuseZ4 opened 10 months ago

ZuseZ4 commented 10 months ago

@bytesnake Enzyme Api recently needed some small breakage anyway, so Billy added the batching along the way. I'll fix it on the rust side and probably add a wrapper for the pure (highly fragile) batching mode without AD, by copying most from the autodiff macro. We should sit together one day to decide how to update our original AD macro to allow vectorized AD.