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