traveler.tables

JAX-backed datasets and tables with pandas and Polars conversion.

Functions

consolidate(df)	Consolidate Polars data to ensure all columns are a single chunk of memory.
jax_to_polars(table[, include_prng_key])	Convert a JaxTable to a polars DataFrame.
polars_to_jax(df[, cls, prng_keys])	Convert a Polars data frame to a JAX-backed table.

Classes

JaxDataset(*args, **kwargs)	A collection of JAX arrays of with metadata.
JaxTable(*args, **kwargs)	Analog of a pyarrow Table, but with JAX arrays.