# Fashionable Modelling with Flux

arXiv: https://arxiv.org/abs/1811.01457 · HTML: https://mlsanity.com/paper/1811.01457/

## Key facts

- Cited by 53 papers in the map (data through 14 Oct 2026).
- First posted on arXiv 1 Nov 2018.

## Built on by (most cited first, 30 of 53)

- [Equinox: neural networks in JAX via callable PyTrees and filtered transformations](https://mlsanity.com/paper/2111.00254/) (2021)
- [Calibration tests beyond classification](https://mlsanity.com/paper/2210.13355/) (2022)
- [Multiple shooting for training neural differential equations on time series](https://mlsanity.com/paper/2109.06786/) (2021)
- [Stabilized Neural Differential Equations for Learning Dynamics with Explicit Constraints](https://mlsanity.com/paper/2306.09739/) (2023)
- [Machine Learning and Deep Learning -- A review for Ecologists](https://mlsanity.com/paper/2204.05023/) (2022)
- [Dual Lagrangian Learning for Conic Optimization](https://mlsanity.com/paper/2402.03086/) (2024)
- [Endogenous Macrodynamics in Algorithmic Recourse](https://mlsanity.com/paper/2308.08187/) (2023)
- [Machine Learning Kreuzer--Skarke Calabi--Yau Threefolds](https://mlsanity.com/paper/2112.09117/) (2021)
- [Learned multiphysics inversion with differentiable programming and machine learning](https://mlsanity.com/paper/2304.05592/) (2023)
- [Learning Optimal Power Flow Value Functions with Input-Convex Neural Networks](https://mlsanity.com/paper/2310.04605/) (2023)
- [SeaPearl: A Constraint Programming Solver guided by Reinforcement Learning](https://mlsanity.com/paper/2102.09193/) (2021)
- [The Contextual Lasso: Sparse Linear Models via Deep Neural Networks](https://mlsanity.com/paper/2302.00878/) (2023)
- [Deep learning approaches to surrogates for solving the diffusion equation for mechanistic real-world simulations](https://mlsanity.com/paper/2102.05527/) (2021)
- [Multicriteria interpretability driven Deep Learning](https://mlsanity.com/paper/2111.14088/) (2021)
- [Fast Conditional Network Compression Using Bayesian HyperNetworks](https://mlsanity.com/paper/2205.06404/) (2022)
- [What You See is Not What You Get: Neural Partial Differential Equations and The Illusion of Learning](https://mlsanity.com/paper/2411.15101/) (2024)
- [Branching Flows: Discrete, Continuous, and Manifold Flow Matching with Splits and Deletions](https://mlsanity.com/paper/2511.09465/) (2025)
- [Optimal Energy Shaping via Neural Approximators](https://mlsanity.com/paper/2101.05537/) (2021)
- [Learning Modular Simulations for Homogeneous Systems](https://mlsanity.com/paper/2210.16294/) (2022)
- [Global Performance Guarantees for Neural Network Models of AC Power Flow](https://mlsanity.com/paper/2211.07125/) (2022)
- [Analyzing the Performance of Deep Encoder-Decoder Networks as Surrogates for a Diffusion Equation](https://mlsanity.com/paper/2302.03786/) (2023)
- [A Physics-Informed Machine Learning for Electricity Markets: A NYISO Case Study](https://mlsanity.com/paper/2304.00062/) (2023)
- [ProDAG: Projected Variational Inference for Directed Acyclic Graphs](https://mlsanity.com/paper/2405.15167/) (2024)
- [Sparser, Better, Faster, Stronger: Sparsity Detection for Efficient Automatic Differentiation](https://mlsanity.com/paper/2501.17737/) (2025)
- [Semi-Explicit Neural DAEs: Learning Long-Horizon Dynamical Systems with Algebraic Constraints](https://mlsanity.com/paper/2505.20515/) (2025)
- [Mapping the Internet: Modelling Entity Interactions in Complex Heterogeneous Networks](https://mlsanity.com/paper/2104.09650/) (2021)
- [MLPGradientFlow: going with the flow of multilayer perceptrons (and finding minima fast and accurately)](https://mlsanity.com/paper/2301.10638/) (2023)
- [Parameter Inference with Bifurcation Diagrams](https://mlsanity.com/paper/2106.04243/) (2021)
- [STEADY: Simultaneous State Estimation and Dynamics Learning from Indirect Observations](https://mlsanity.com/paper/2203.01299/) (2022)
- [Machines of finite depth: towards a formalization of neural networks](https://mlsanity.com/paper/2204.12786/) (2022)

## Abstract (arXiv)

Machine learning as a discipline has seen an incredible surge of interest in recent years due in large part to a perfect storm of new theory, superior tooling, renewed interest in its capabilities. We present in this paper a framework named Flux that shows how further refinement of the core ideas of machine learning, built upon the foundation of the Julia programming language, can yield an environment that is simple, easily modifiable, and performant. We detail the fundamental principles of Flux as a framework for differentiable programming, give examples of models that are implemented within Flux to display many of the language and framework-level features that contribute to its ease of use and high productivity, display internal compiler techniques used to enable the acceleration and performance that lies at the heart of Flux, and finally give an overview of the larger ecosystem that Flux fits inside of.