https://www.glaciology.net/tags/jax/ Recent content in JAX on Aslak Grinsted Hugo en-us Sat, 24 Oct 2026 00:00:00 +0000 https://www.glaciology.net/studentproject/2026/jax_icesheet/ Sat, 24 Oct 2026 00:00:00 +0000 https://www.glaciology.net/studentproject/2026/jax_icesheet/ <p>Ice sheets evolve over centuries to millennia, yet most ice-sheet models are forced to take time steps of only a few years to remain numerically stable. Can we do better? And can we build ice-flow models that are fast, run on GPUs, and are differentiable, so that they can be calibrated with gradients and coupled into modern climate models?</p> <p>In this computational project, you will develop a compact ice-flow model in JAX, starting from the Shallow Shelf Approximation in a 1D flowline setting and possibly extending to 2D. The model will be verified against analytical solutions and standard benchmarks such as MISMIP and ISMIP-HOM. Using automatic differentiation, you will compute gradients of model output with respect to inputs like basal friction and surface mass balance.</p>