Flux

Flux is an open-source machine learning library fully written in Julia, designed for researchers, developers, and data scientists seeking a lightweight, highly hackable ML stack. It offers seamless GPU acceleration, differentiable programming, and easy-to-use model definitions using Julia’s mathematical syntax, while remaining highly extensible for advanced users and use cases. Flux is central to the Julia ML ecosystem, supporting deep learning, probabilistic programming, graph neural networks, computer vision, natural language processing, and more.