traveler.demo.work_location.JaxTable.to_dict# JaxTable.to_dict()[source]# 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]
Traveler