biogeme.expressions.jax_utils module

Define various items for Jax

Michel Bierlaire Tue Mar 18 18:28:07 2025

biogeme.expressions.jax_utils.build_vectorized_function(the_function, use_jit)[source]

Build the function that is applied to each row of the databaser

Parameters:

use_jit (bool)