DiffEqFlux.jl icon indicating copy to clipboard operation
DiffEqFlux.jl copied to clipboard

Restructuring some of the Layers

Open avik-pal opened this issue 1 year ago • 2 comments

Question❓

What do we feel about moving some of the non-diffeq specific layers to Boltz.jl and re-exporting them from here?

I was working on implementing KANs and we might be able to reuse the Spline Layers from here, but DiffEqFlux has too heavy deps (not complaining here, it needs to). Boltz.jl on the other hand is quite lightweight, only pulling in stdlibs and Lux (and some other smallish packages).

avik-pal avatar Jun 08 '24 19:06 avik-pal

Yeah go for it.

ChrisRackauckas avatar Jun 08 '24 20:06 ChrisRackauckas

Most layers are now part of Boltz. Spline Layer will be added in https://github.com/LuxDL/Boltz.jl/pull/35.

avik-pal avatar Jun 09 '24 21:06 avik-pal