Modern, Efficient, and Differentiable Transport Equation Models using JAX: Applications to Population Balance Equations. - Mohammed Alsubeihi; Arthur Jessop; Ben Moseley; Cláudio P. Fonte; Ashwin Kumar Rajagopalan (CoRR, 2024)