search

LuxDL / Lux.jl

Elegant & Performant Scientific Machine Learning in Julia
https://lux.csail.mit.edu/
MIT License
456 stars 54 forks source link

Add a compiled tape version for ReverseDiff #642

Closed avik-pal closed 4 weeks ago

avik-pal commented 2 months ago

With the new caching API introduced in https://github.com/LuxDL/Lux.jl/pull/640 this shouldn't be too hard