AMJAX Jan 1, 2026 Go to Project Site Scientific computing Scientific software Partial differential equations JAX Scientific machine learning Victor Boussange Applied ML researcher AI for science