Update jax, make fields available on a global level to models (type_id = 0)