Flux v0.3 is a big release, with a brand new, pure-Julia backend for machine learning. This lets us immediately support all of Julia’s most powerful features, without having to compile them to a more limited backend like MXNet. It’s also a huge simplification, with a codebase that’s half the lines and much more straightforward.
Going forward, it will be much easier for us to apply our own optimisations and integrate with Julia’s rapidly growing GPU ecosystem. The housing data example shows how easy it is to drop CuArrays (or GPUArrays.jl) into a model for an easy speedup, with more integration coming shortly.
Of course, you don’t need a GPU to get started, and trying it out is as simple as
Check out the new docs and enjoy!