scispace - formally typeset
Journal ArticleDOI

Algorithm 799: revolve: an implementation of checkpointing for the reverse or adjoint mode of computational differentiation

TLDR
This article presents the function revolve, which generates checkpointing schedules that are provably optimal with regard to a primary and a secondary criterion and is intended to be used as an explicit “controller” for running a time-dependent applications program.
Abstract
In its basic form, the reverse mode of computational differentiation yields the gradient of a scalar-valued function at a cost that is a small multiple of the computational work needed to evaluate the function itself. However, the corresponding memory requirement is proportional to the run-time of the evaluation program. Therefore, the practical applicability of the reverse mode in its original formulation is limited despite the availability of ever larger memory systems. This observation leads to the development of checkpointing schedules to reduce the storage requirements. This article presents the function revolve, which generates checkpointing schedules that are provably optimal with regard to a primary and a secondary criterion. This routine is intended to be used as an explicit “controller” for running a time-dependent applications program.

read more

Citations
More filters
Journal ArticleDOI

Transparent Checkpointing for Automatic Differentiation of Program Loops Through Expression Transformations

TL;DR: Checkpointing as discussed by the authors leverages expression transformations in the programming language Julia and the package ChainRules.jl to automatically and transparently transform loop iterations into differentiated loops, and demonstrate its features on an automatically differentiated MPI implementation of Burgers' equation on the Polaris cluster at the Argonne Leadership Computing Facility.
Posted Content

Efficient Large-Scale Language Model Training on GPU Clusters Using Megatron-LM

TL;DR: In this article, different types of parallelism methods (tensor, pipeline, and data parallelism) can be composed to scale to thousands of GPUs and models with trillions of parameters, and a novel interleaved pipeline parallelism schedule that can improve throughput by 10+% with memory footprint comparable to existing approaches.
Journal ArticleDOI

Nonlinear optimal perturbations and formation mechanism of localized wave packet in channel flow

- 01 May 2023 - 
TL;DR: In this paper , a nonlinear nonmodal optimization method is used to find the minimal energy perturbation triggering the subcritical transition in a two-dimensional channel flow is characterized by the emergence of localized wave packet (LWP).
Proceedings ArticleDOI

ScaDLES: Scalable Deep Learning over Streaming data at the Edge

TL;DR: ScaDLES as mentioned in this paper proposes to train on streaming data at the edge in an online fashion, while also addressing the challenges of limited bandwidth and training with non-IID data.
Proceedings ArticleDOI

Partitioned solution of the unsteady adjoint equations for the one-dimensional flowin a flexible tube

TL;DR: In this article, the gradient of an objective function for an elastic tube through which an incompressible flow of an inviscid fluid flows through the tube is derived and solved using quasi-Newton coupling iterations.
References
More filters
Book

Numerical methods for conservation laws

TL;DR: In this paper, the authors describe the derivation of conservation laws and apply them to linear systems, including the linear advection equation, the Euler equation, and the Riemann problem.
Book

Optimal Control of Systems Governed by Partial Differential Equations

TL;DR: In this paper, the authors consider the problem of minimizing the sum of a differentiable and non-differentiable function in the context of a system governed by a Dirichlet problem.
Book

Evaluating Derivatives: Principles and Techniques of Algorithmic Differentiation

TL;DR: This second edition has been updated and expanded to cover recent developments in applications and theory, including an elegant NP completeness argument by Uwe Naumann and a brief introduction to scarcity, a generalization of sparsity.
Journal ArticleDOI

Upwind difference schemes for hyperbolic systems of conservation laws

TL;DR: In this article, a new upwind finite difference approximation to systems of nonlinear hyperbolic conservation laws has been derived. But the scheme has desirable properties for shock calculations, such as unique and sharp shocks.
Journal ArticleDOI

Achieving logarithmic growth of temporal and spatial complexity in reverse automatic differentiation

TL;DR: It is shown here that, by a recursive scheme related to the multilevel differentiation approach of Volin and Ostrovskii, the growth in both temporal and spatial complexity can be limited to a fixed multiple of log(T).
Related Papers (5)