traveler.trees.JaxDataset.to_dict# JaxDataset.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