traveler.templates.land_use_taz.JaxTable.to_dict

JaxTable.to_dict()

Convert the JaxDataset to a dictionary of JAX arrays.

CategoricalArrays are converted to pandas Series instead of JAX arrays.

Return type:

dict[str, jt.Array | pd.Series]