TensorFlow Eager: A Multi-Stage, Python-Embedded DSL for Machine Learning
Not on the map. Why?
Abstract (arXiv)
TensorFlow Eager is a multi-stage, Python-embedded domain-specific language for hardware-accelerated machine learning, suitable for both interactive research and production. TensorFlow, which TensorFlow Eager extends, requires users to represent computations as dataflow graphs; this permits compiler optimizations and simplifies deployment but hinders rapid prototyping and run-time dynamism. TensorFlow Eager eliminates these usability costs without sacrificing the benefits furnished by graphs: It provides an imperative front-end to TensorFlow that executes operations immediately and a JIT tracer that translates Python functions composed of TensorFlow operations into executable dataflow graphs. TensorFlow Eager thus offers a multi-stage programming model that makes it easy to interpolate between imperative and staged execution in a single package.
Key facts. Cited by 22 papers in the map (data through 7 Oct 2026). First posted on arXiv 27 Feb 2019.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 8; 2022: 6; 2023: 1; 2024: 1; 2025: 6.
22 papers.
Sort:
- 2021CrypTen: Secure Multi-Party Computation Meets Machine Learningcited by 131
- 2022Tutorial on amortized optimizationcited by 83
- 2022Pathways: Asynchronous Distributed Dataflow for MLcited by 81
- 2021Torch.fx: Practical Program Capture and Transformation for Deep Learning in Pythoncited by 38
- 2021DKM: Differentiable K-Means Clustering Layer for Neural Network Compressioncited by 31
- 2022A domain-specific language for describing machine learning datasetscited by 12
- 2021LazyTensor: combining eager execution with domain-specific compilerscited by 11
- 2022Closed-Form Diffeomorphic Transformations for Time Series Alignmentcited by 7
- 2025Quantum Federated Learning: A Comprehensive Surveycited by 7
- 2021Swift for TensorFlow: A portable, flexible platform for deep learningcited by 6
- 2024Bundle Adjustment in the Eager Modecited by 6
- 2021TensorGP -- Genetic Programming Engine in TensorFlowcited by 5
10 more
- 2021Speed Benchmarking of Genetic Programming Frameworkscited by 2
- 2022Challenges in Migrating Imperative Deep Learning Programs to Graph Execution: An Empirical Studycited by 2
- 2025Data-Driven Adaptive Gradient Recovery for Unstructured Finite Volume Computationscited by 2
- 2022Terra: Imperative-Symbolic Co-Execution of Imperative Deep Learning Programscited by 1
- 2025Tempo: Compiled Dynamic Deep Learning with Symbolic Dependence Graphscited by 1
- 2025veScale: Consistent and Efficient Tensor Programming with Eager-Mode SPMDcited by 1
- 2025GraphMend: Code Transformations for Fixing Graph Breaks in PyTorch 2cited by 1
- 2021CLAIMED, a visual and scalable component library for Trusted AIcited by 0
- 2023RAMario: Experimental Approach to Reptile Algorithm -- Reinforcement Learning for Mariocited by 0
- 2025Speculative Automated Refactoring of Imperative Deep Learning Programs to Graph Executioncited by 0
- 2025GraphMend: Code Transformations for Fixing Graph Breaks in PyTorch 217 Sep 2025
- 2025veScale: Consistent and Efficient Tensor Programming with Eager-Mode SPMD5 Sep 2025
- 2025Quantum Federated Learning: A Comprehensive Survey21 Aug 2025
- 2025Data-Driven Adaptive Gradient Recovery for Unstructured Finite Volume Computations22 Jul 2025
- 2025Speculative Automated Refactoring of Imperative Deep Learning Programs to Graph Execution7 Apr 2025
- 2025Tempo: Compiled Dynamic Deep Learning with Symbolic Dependence Graphs9 Jan 2025
- 2024Bundle Adjustment in the Eager Mode18 Sep 2024
- 2023RAMario: Experimental Approach to Reptile Algorithm -- Reinforcement Learning for Mario16 May 2023
- 2022A domain-specific language for describing machine learning datasets5 Jul 2022
- 2022Closed-Form Diffeomorphic Transformations for Time Series Alignment16 Jun 2022
- 2022Pathways: Asynchronous Distributed Dataflow for ML23 Mar 2022
- 2022Tutorial on amortized optimization1 Feb 2022
10 more
- 2022Challenges in Migrating Imperative Deep Learning Programs to Graph Execution: An Empirical Study24 Jan 2022
- 2022Terra: Imperative-Symbolic Co-Execution of Imperative Deep Learning Programs23 Jan 2022
- 2021Torch.fx: Practical Program Capture and Transformation for Deep Learning in Python15 Dec 2021
- 2021CrypTen: Secure Multi-Party Computation Meets Machine Learning2 Sep 2021
- 2021DKM: Differentiable K-Means Clustering Layer for Neural Network Compression28 Aug 2021
- 2021Speed Benchmarking of Genetic Programming Frameworks25 May 2021
- 2021TensorGP -- Genetic Programming Engine in TensorFlow12 Mar 2021
- 2021CLAIMED, a visual and scalable component library for Trusted AI4 Mar 2021
- 2021LazyTensor: combining eager execution with domain-specific compilers26 Feb 2021
- 2021Swift for TensorFlow: A portable, flexible platform for deep learning26 Feb 2021
- 2021CrypTen: Secure Multi-Party Computation Meets Machine Learningcited by 131
- 2022Tutorial on amortized optimizationcited by 83
- 2022Pathways: Asynchronous Distributed Dataflow for MLcited by 81
- 2021Torch.fx: Practical Program Capture and Transformation for Deep Learning in Pythoncited by 38
- 2021DKM: Differentiable K-Means Clustering Layer for Neural Network Compressioncited by 31
- 2022A domain-specific language for describing machine learning datasetscited by 12
- 2021LazyTensor: combining eager execution with domain-specific compilerscited by 11
- 2022Closed-Form Diffeomorphic Transformations for Time Series Alignmentcited by 7
- 2025Quantum Federated Learning: A Comprehensive Surveycited by 7
- 2021Swift for TensorFlow: A portable, flexible platform for deep learningcited by 6
- 2024Bundle Adjustment in the Eager Modecited by 6
- 2021TensorGP -- Genetic Programming Engine in TensorFlowcited by 5
12 most cited of 22 All 22 (JSON)
- 2025GraphMend: Code Transformations for Fixing Graph Breaks in PyTorch 217 Sep 2025
- 2025veScale: Consistent and Efficient Tensor Programming with Eager-Mode SPMD5 Sep 2025
- 2025Quantum Federated Learning: A Comprehensive Survey21 Aug 2025
- 2025Data-Driven Adaptive Gradient Recovery for Unstructured Finite Volume Computations22 Jul 2025
- 2025Speculative Automated Refactoring of Imperative Deep Learning Programs to Graph Execution7 Apr 2025
- 2025Tempo: Compiled Dynamic Deep Learning with Symbolic Dependence Graphs9 Jan 2025
- 2024Bundle Adjustment in the Eager Mode18 Sep 2024
- 2023RAMario: Experimental Approach to Reptile Algorithm -- Reinforcement Learning for Mario16 May 2023
- 2022A domain-specific language for describing machine learning datasets5 Jul 2022
- 2022Closed-Form Diffeomorphic Transformations for Time Series Alignment16 Jun 2022
- 2022Pathways: Asynchronous Distributed Dataflow for ML23 Mar 2022
- 2022Tutorial on amortized optimization1 Feb 2022
12 newest of 22 All 22 (JSON)