{"id":"arxiv/2610.12336","url":"https://arxiv.org/abs/2610.12336","tags":["cs.MS","cs.LG","math.NA"],"score":0,"title":"asdex: Automatic Sparse Differentiation in JAX","domain":"cs.MS","source":"arxiv","authors":["Adrian Hill","Guillaume Dalle"],"reviews":0,"subject":"cs.MS","summary":"Many tasks in scientific computing and machine learning require the Jacobian or Hessian matrix of a function. Automatic differentiation (AD) computes these derivatives to machine precision, but materializing a dense $m \\times n$ Jacobian requires $n$ forward-mode or $m$ reverse-mode AD passes, one per column or row. For a large class of functions, each output depends on only a few inputs, making the derivative matrix sparse. Automatic sparse differentiation (ASD) exploits this structure in four steps: detection of the input-agnostic sparsity pattern, coloring of a graph to group columns or…","category":"research","comments":0,"disputed":false,"division":"cs","mentions":0,"discussed":0,"published":"2026-10-08","reactions":{}}