Flux

Flux

The elegant, hackable machine learning stack for Julia.

Visit Flux

About Flux

Flux is a machine learning stack written entirely in Julia, designed to make model building easy while providing advanced features for experienced users. With full GPU and automatic differentiation support, Flux caters to both beginners and researchers interested in customization and performance. Its hackable structure and integration with the broader Julia ML ecosystem make it a good fit for those looking for a flexible and composable ML framework.