Compiled Eager Code
Flux provides a single, intuitive way to define models, just like mathematical notation. Julia transparently compiles your code
, optimising and fusing kernels for the GPU, for the best performance.
Existing Julia libraries are differentiable and can be incorporated directly into Flux models. Cutting edge models such as Neural ODEs
are first class, and Zygote
enables overhead-free gradients.
First-class GPU support
GPU kernels can be written directly in Julia via CUDA.jl
. Flux is uniquely hackable and any part can be tweaked, from GPU code to custom gradients and layers.