traveler.tables.jax_to_polars#

traveler.tables.jax_to_polars(table, include_prng_key=True)[source]#

Convert a JaxTable to a polars DataFrame.

Parameters:

  • table (JaxTable) – The JaxTable to convert.

  • include_prng_key (bool)

Returns:

A polars representation of the JaxTable.

Return type:

polars.DataFrame