Fashionable Modelling with Flux
Not on the map. Why?
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.
Key facts. Cited by 53 papers in the map (data through 14 Oct 2026). First posted on arXiv 1 Nov 2018.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 14; 2022: 11; 2023: 14; 2024: 5; 2025: 6; 2026: 3.
53 papers.
Sort:
By area:
- 2021Equinox: neural networks in JAX via callable PyTrees and filtered transformationscited by 148
- 2022Calibration tests beyond classificationcited by 20
- 2021Multiple shooting for training neural differential equations on time seriescited by 17
- 2023Stabilized Neural Differential Equations for Learning Dynamics with Explicit Constraintscited by 15
- 2022Machine Learning and Deep Learning -- A review for Ecologistscited by 11
- 2024Dual Lagrangian Learning for Conic Optimizationcited by 11
- 2023Endogenous Macrodynamics in Algorithmic Recoursecited by 9
- 2021Machine Learning Kreuzer--Skarke Calabi--Yau Threefoldscited by 8
- 2023Learned multiphysics inversion with differentiable programming and machine learningcited by 7
- 2023Learning Optimal Power Flow Value Functions with Input-Convex Neural Networkscited by 7
- 2021SeaPearl: A Constraint Programming Solver guided by Reinforcement Learningcited by 6
- 2023The Contextual Lasso: Sparse Linear Models via Deep Neural Networkscited by 6
38 more
- 2021Deep learning approaches to surrogates for solving the diffusion equation for mechanistic real-world simulationscited by 5
- 2021Multicriteria interpretability driven Deep Learningcited by 5
- 2022Fast Conditional Network Compression Using Bayesian HyperNetworkscited by 5
- 2024What You See is Not What You Get: Neural Partial Differential Equations and The Illusion of Learningcited by 5
- 2025Branching Flows: Discrete, Continuous, and Manifold Flow Matching with Splits and Deletionscited by 5
- 2021Optimal Energy Shaping via Neural Approximatorscited by 4
- 2022Learning Modular Simulations for Homogeneous Systemscited by 4
- 2022Global Performance Guarantees for Neural Network Models of AC Power Flowcited by 4
- 2023Analyzing the Performance of Deep Encoder-Decoder Networks as Surrogates for a Diffusion Equationcited by 4
- 2023A Physics-Informed Machine Learning for Electricity Markets: A NYISO Case Studycited by 4
- 2024ProDAG: Projected Variational Inference for Directed Acyclic Graphscited by 4
- 2025Sparser, Better, Faster, Stronger: Sparsity Detection for Efficient Automatic Differentiationcited by 4
- 2025Semi-Explicit Neural DAEs: Learning Long-Horizon Dynamical Systems with Algebraic Constraintscited by 4
- 2021Mapping the Internet: Modelling Entity Interactions in Complex Heterogeneous Networkscited by 3
- 2023MLPGradientFlow: going with the flow of multilayer perceptrons (and finding minima fast and accurately)cited by 3
- 2021Parameter Inference with Bifurcation Diagramscited by 2
- 2022STEADY: Simultaneous State Estimation and Dynamics Learning from Indirect Observationscited by 2
- 2022Machines of finite depth: towards a formalization of neural networkscited by 2
- 2022Recursive Neural Programs: Variational Learning of Image Grammars and Part-Whole Hierarchiescited by 2
- 2022Training neural network ensembles via trajectory samplingcited by 2
- 2023Effective Latent Differential Equation Models via Attention and Multiple Shootingcited by 2
- 2024Dirac--Bianconi Graph Neural Networks -- Enabling Non-Diffusive Long-Range Graph Predictionscited by 2
- 2021Optimal Transport Based Refinement of Physics-Informed Neural Networkscited by 1
- 2021Physics informed machine learning with Smoothed Particle Hydrodynamics: Hierarchy of reduced Lagrangian models of turbulencecited by 1
- 2022SAFE-OCC: A Novelty Detection Framework for Convolutional Neural Network Sensors and its Application in Process Controlcited by 1
- 2022Deep learning and differential equations for modeling changes in individual-level latent dynamics between observation periodscited by 1
- 2023Random Function Descentcited by 1
- 2025Adaptive Preference Aggregationcited by 1
- 2021EBIC.JL -- an Efficient Implementation of Evolutionary Biclustering Algorithm in Juliacited by 0
- 2021Chameleon: A Semi-AutoML framework targeting quick and scalable development and deployment of production-ready ML systems for SMEscited by 0
- 2021XConv: Low-memory stochastic backpropagation for convolutional layerscited by 0
- 2023Chemellia: An Ecosystem for Atomistic Scientific Machine Learningcited by 0
- 2023Kernelised Normalising Flowscited by 0
- 2023Deep Learning for Koopman-based Dynamic Movement Primitivescited by 0
- 2023Taming Waves: A Physically-Interpretable Machine Learning Framework for Realizable Control of Wave Dynamicscited by 0
- 2024A Physics-Constrained Neural Differential Equation Framework for Data-Driven Snowpack Simulationcited by 0
- 2025Residual-Informed Learning of Solutions to Algebraic Loopscited by 0
- 2025Unified Implementations of Recurrent Neural Networks in Multiple Deep Learning Frameworkscited by 0
50 most cited of 53 All 53 (JSON)
- 2026Low-Budget Active Learning through Entropic Optimal Transport1 Oct 2026
- 2026Accelerated Learning of High Dimensional Functions with a Tensor-Featured Training Network11 Aug 2026
- 2026DisjunctiveNet: Neural Symbolic Learning via Differentiable Convexified Optimization Layers28 May 2026
- 2025Branching Flows: Discrete, Continuous, and Manifold Flow Matching with Splits and Deletions12 Nov 2025
- 2025Unified Implementations of Recurrent Neural Networks in Multiple Deep Learning Frameworks24 Oct 2025
- 2025Residual-Informed Learning of Solutions to Algebraic Loops10 Oct 2025
- 2025Semi-Explicit Neural DAEs: Learning Long-Horizon Dynamical Systems with Algebraic Constraints26 May 2025
- 2025Adaptive Preference Aggregation13 Mar 2025
- 2025Sparser, Better, Faster, Stronger: Sparsity Detection for Efficient Automatic Differentiation29 Jan 2025
- 2024A Physics-Constrained Neural Differential Equation Framework for Data-Driven Snowpack Simulation3 Dec 2024
- 2024What You See is Not What You Get: Neural Partial Differential Equations and The Illusion of Learning22 Nov 2024
- 2024Dirac--Bianconi Graph Neural Networks -- Enabling Non-Diffusive Long-Range Graph Predictions17 Jul 2024
38 more
- 2024ProDAG: Projected Variational Inference for Directed Acyclic Graphs24 May 2024
- 2024Dual Lagrangian Learning for Conic Optimization5 Feb 2024
- 2023Deep Learning for Koopman-based Dynamic Movement Primitives6 Dec 2023
- 2023Taming Waves: A Physically-Interpretable Machine Learning Framework for Realizable Control of Wave Dynamics27 Nov 2023
- 2023Learning Optimal Power Flow Value Functions with Input-Convex Neural Networks6 Oct 2023
- 2023Endogenous Macrodynamics in Algorithmic Recourse16 Aug 2023
- 2023Kernelised Normalising Flows27 Jul 2023
- 2023Effective Latent Differential Equation Models via Attention and Multiple Shooting11 Jul 2023
- 2023Stabilized Neural Differential Equations for Learning Dynamics with Explicit Constraints16 Jun 2023
- 2023Chemellia: An Ecosystem for Atomistic Scientific Machine Learning19 May 2023
- 2023Random Function Descent2 May 2023
- 2023Learned multiphysics inversion with differentiable programming and machine learning12 Apr 2023
- 2023A Physics-Informed Machine Learning for Electricity Markets: A NYISO Case Study31 Mar 2023
- 2023Analyzing the Performance of Deep Encoder-Decoder Networks as Surrogates for a Diffusion Equation7 Feb 2023
- 2023The Contextual Lasso: Sparse Linear Models via Deep Neural Networks2 Feb 2023
- 2023MLPGradientFlow: going with the flow of multilayer perceptrons (and finding minima fast and accurately)25 Jan 2023
- 2022Global Performance Guarantees for Neural Network Models of AC Power Flow14 Nov 2022
- 2022Learning Modular Simulations for Homogeneous Systems28 Oct 2022
- 2022Calibration tests beyond classification21 Oct 2022
- 2022Training neural network ensembles via trajectory sampling22 Sep 2022
- 2022Recursive Neural Programs: Variational Learning of Image Grammars and Part-Whole Hierarchies16 Jun 2022
- 2022Fast Conditional Network Compression Using Bayesian HyperNetworks13 May 2022
- 2022Machines of finite depth: towards a formalization of neural networks27 Apr 2022
- 2022Machine Learning and Deep Learning -- A review for Ecologists11 Apr 2022
- 2022STEADY: Simultaneous State Estimation and Dynamics Learning from Indirect Observations2 Mar 2022
- 2022Deep learning and differential equations for modeling changes in individual-level latent dynamics between observation periods15 Feb 2022
- 2022SAFE-OCC: A Novelty Detection Framework for Convolutional Neural Network Sensors and its Application in Process Control3 Feb 2022
- 2021Machine Learning Kreuzer--Skarke Calabi--Yau Threefolds16 Dec 2021
- 2021Multicriteria interpretability driven Deep Learning28 Nov 2021
- 2021Equinox: neural networks in JAX via callable PyTrees and filtered transformations30 Oct 2021
- 2021Physics informed machine learning with Smoothed Particle Hydrodynamics: Hierarchy of reduced Lagrangian models of turbulence25 Oct 2021
- 2021Multiple shooting for training neural differential equations on time series14 Sep 2021
- 2021XConv: Low-memory stochastic backpropagation for convolutional layers13 Jun 2021
- 2021Parameter Inference with Bifurcation Diagrams8 Jun 2021
- 2021Optimal Transport Based Refinement of Physics-Informed Neural Networks26 May 2021
- 2021Chameleon: A Semi-AutoML framework targeting quick and scalable development and deployment of production-ready ML systems for SMEs8 May 2021
- 2021EBIC.JL -- an Efficient Implementation of Evolutionary Biclustering Algorithm in Julia3 May 2021
- 2021Mapping the Internet: Modelling Entity Interactions in Complex Heterogeneous Networks19 Apr 2021
50 newest of 53 All 53 (JSON)
- 2021Equinox: neural networks in JAX via callable PyTrees and filtered transformationscited by 148
- 2022Calibration tests beyond classificationcited by 20
- 2021Multiple shooting for training neural differential equations on time seriescited by 17
- 2023Stabilized Neural Differential Equations for Learning Dynamics with Explicit Constraintscited by 15
- 2022Machine Learning and Deep Learning -- A review for Ecologistscited by 11
- 2024Dual Lagrangian Learning for Conic Optimizationcited by 11
- 2023Endogenous Macrodynamics in Algorithmic Recoursecited by 9
- 2021Machine Learning Kreuzer--Skarke Calabi--Yau Threefoldscited by 8
- 2023Learned multiphysics inversion with differentiable programming and machine learningcited by 7
- 2023Learning Optimal Power Flow Value Functions with Input-Convex Neural Networkscited by 7
- 2021SeaPearl: A Constraint Programming Solver guided by Reinforcement Learningcited by 6
- 2023The Contextual Lasso: Sparse Linear Models via Deep Neural Networkscited by 6
12 most cited of 52 All 52 (JSON)
- 2026Low-Budget Active Learning through Entropic Optimal Transport1 Oct 2026
- 2026Accelerated Learning of High Dimensional Functions with a Tensor-Featured Training Network11 Aug 2026
- 2026DisjunctiveNet: Neural Symbolic Learning via Differentiable Convexified Optimization Layers28 May 2026
- 2025Unified Implementations of Recurrent Neural Networks in Multiple Deep Learning Frameworks24 Oct 2025
- 2025Residual-Informed Learning of Solutions to Algebraic Loops10 Oct 2025
- 2025Semi-Explicit Neural DAEs: Learning Long-Horizon Dynamical Systems with Algebraic Constraints26 May 2025
- 2025Adaptive Preference Aggregation13 Mar 2025
- 2025Sparser, Better, Faster, Stronger: Sparsity Detection for Efficient Automatic Differentiation29 Jan 2025
- 2024A Physics-Constrained Neural Differential Equation Framework for Data-Driven Snowpack Simulation3 Dec 2024
- 2024What You See is Not What You Get: Neural Partial Differential Equations and The Illusion of Learning22 Nov 2024
- 2024Dirac--Bianconi Graph Neural Networks -- Enabling Non-Diffusive Long-Range Graph Predictions17 Jul 2024
- 2024ProDAG: Projected Variational Inference for Directed Acyclic Graphs24 May 2024
12 newest of 52 All 52 (JSON)