Preprint
Sep 2026
TrunX: A massively parallel, differentiable implementation of the 3-PG forest growth model in JAX
An implementation of the Physiological Processes Predicting Growth (3-PG) model in JAX that uses just-in-time compilation, vectorization, and GPU acceleration to reduce execution time and enables efficient gradient-based optimization and gradient-informed Bayesian calibration.
G. M. Givi, Cédric Travelletti, Grégory Mermoud
· 0 citations